Added functionality to game over screen.