Implementing a decision tree with scikit-learn