hw2b.py added