adding a unit test and minor modifications
added junit to the class path