Create decision tree.py