More improvements to python exercise
Finished user input section of exercise.