Added better game over flow