Changes to integrate a data provider for TestNG