Update hw2b.py