Use junit instead of testng