Fixing free state, and making game over and end screens lead to main screen.