JUnit import changed to TestNG