Update exercise3.py