Entanglement entropy dynamics of Heisenberg chains