Add tests for problem 4
Solution for problem 6
Remove extraneous println from problem 2