adding some junit tests