Update Homework2B.py