finish desicion tree, but need to know more about py data type and matplotlib