Problem 2 is done