Add sklearn requirement