Fixed the tree coloring issue.