Change some end game logic