modified:   hw2.py