Update main_tree.cpp