Go back to looking at `stats_to_tree_dict`