add some code to main.cpp (arbre)