Add endGame and stuff