Add endgame screen. Fixes #57