removed some junit imports and added testng for it.