Add custom tree functionnality