Added screenshot support for TestNG