Add junit tests