remove branchInfo and depth for coloring. Not yet completed.