added extra #yourcodehere in Prob 1(a) in HW2.ipynb