added hw2b.py