Update exercise2.py