updated the junit tests