Completed homework from week 4