Changed QTreeWidget to QTreeView