Use TestNG instead of JUnit