Bug with end of game fixed