Updated to JUnit4