add hw2 problem