Added option to change color of the tree