just adding the animation code