Create hw2p1.py