Added Junit test for problem 1