more treeview extension code

Change-Id: I12262783e98aa342069c94ee8a7dcdab1fb51ee7