Finally finished changes for homework

fixed changes to the homework for week 4