Decision tree python code from scratch