Improve end-game animation