Add hw2a.py