Add JUnit 4 to build script