Create HW_4_Prob2.py