Updated from Junit to TestNG