Added Junit codes