Added legend to tree display

added legend to the tree display, although would be perhaps be better as
a separate panel. Will consider it with supervisor