Update homework.py