updating junit tests