Add tests to the project.

**Is your feature request related to a problem? Please describe.**
There are no tests in the project yet.

**Describe the solution you'd like**
A few test cases with JUnit.
