Switch test framework to testng