Remove unneded import
Check dataset privacy in a better way