Added Junit Testing