Decision Tree Data Structure Python