Updating Junit to testNG