a small modification of Week1_Lecture1.ipynb