Added support for undotree