Fixed some of the animation