for Week4 homework