Pruning Decision Tree Python