[python] How do I restart a program based on user input?