Update homework2.py