Add JUnit test suite