Update homework1.py