Implemented JUnit tests [#10]