adding class to testNg file