Add TestNG support
WIP WIP