better handling of game over