add hw3.py