Switching from TestNG to JUnit