Python Decision Tree Algorithm