Added sklearn dependence