tree_compare theme updated to work with changes in the api
key in tree_annot can accept a callback instead of a string