new file:   2hw/hw2.py