Removed sklearn dependencies