Decision Tree exercise by Python