JUnit3 test updated to JUnit4