Update homework4.py