Removed unused TestNG imports