Update decision tree.py