added tests

JUnit test added