add hw2.py