Homework_1_Prob_1_GuoweiCai.ipynb modified