+ fix the animation function