Use JUnit instead of TestNG