Adds support for gen_tree highlighting.