Update HW_3_pt2.py