Added a helper extension for TreeView