Made into TestNG test