Added Junit test