Create HW_3_pt2.py