end of game screen works now