Fixed game over screen, other minor changes