Update exercise6.py