Update exercise 5.py