Add TestNG support