Update HW 5

adding start of problem 3