Added an import for sklearn