modified week2 homework