Problem 2 done.