Update week5.py