Better way to handle game end.