Adding function to color clusters in tree.