Easy Python Decision Tree