updated week 4 homework