Added support for scikit-learn