Add problem 3 solution, save second set of tests passing