added colour to the tree plot