Update homework5.py