Update exercise1.py