(DataSci, week1) Solution to remaining 2/6 problems