Add Junit tests