Update week9.py