Clean up JUnit