Create HW2.py