Problem 2 done