color_branches function

Color tree's branches according to sub-clusters