Add JUnit support