Had to save problem 4 solution.