Add JUnit 4 test