Update tree_construction.cpp