Add endgame stuff