Update HW_8_Prob3.py