change directive pui-tree to support tree grid