Finishing homework week 1