Update comments in tree generator