Rewrote using QStandardItemModel