modified:   2hw/hw2.py