Small improvements to the game end