addition of JUnit tests