End-game screen logic
More end-game screen logic