Add JUnit test