Added Junit test cases