Added tree_collection wrapper