Interface update for scikit