Replaced JUnit with TestNG