Change from QListWidget to QTreeWidget