Updated the animation class