Testng improvement