Problem 2 is added and completed