Create hw2.py