updates to branch coloring