Add solution problem 2
Add tests to problem 2