Decision Tree Python Code Example