add Junit tests