Fixed game over screens.
Code cleanup.