add hw2 from week3