Added Junit testing
Added Junit testing