Add minor changes for end of game