Fixed some treeview stuff to work with icon view