Tidying up the animation