Updated test class in testng.xml files