Move pygame code into run_game()