refactor start screen and game over screen functionality