add JUnit tests