Added extra JUnit