Add tree generator