Tweaks to improve the return animation - comments