Update exercise.py