Update IMN430-week3.py