Add reset animation function, to reset the animation if needed