Decision tree classification with scikit-learn