Minor tweaks to the animation script