Fixed bugs in the game over screen