- Tweaked the animation code