Problem 2 solved