Add Problem 4 Solution