Update homework3.py