added some junit tests