Update treegen.cpp