Improved the animation