problem 2 done