Add tree branches in stage1