Update exercise4.py