code for the animation