Problem 2 done!