Added first attempt at overriding collapse behavior for TreeView.