Finishing homework for week 2