Add: junit tests