Create decision_tree.py