Update python_exercise.py