Update Homework3B.py