Just solved Problem 4