JUnit imports updated