(Datasci, week2) Solutions for week 2 assignment