371 Decision tree for python 3