Add HW2 prob1