1 files changed, 5 insertions, 3 deletions
diff --git a/src/dialog.h b/src/dialog.h
index 17d5cd4..49c3bb0 100644
--- a/src/dialog.h
+++ b/src/dialog.h
@@ -45,24 +45,26 @@ class Dialog : public QDialog {
 
   private: // Private vars n methods
     Ui::Dialog *ui;
-    SessionTreeModel *model_[3]; // TODO: Constants/Enum for indices
-    QPushButton *tabs_[3];
+    SessionTreeModel *model_[TAB_COUNT];
+    QPushButton *tabs_[TAB_COUNT];
     QPoint oldCenter_;
     QTimer *centerTimer_;
     QString autoStartEntry_;
     int activeTab_;
     int defaultTab_;
     bool userInteracted_;
+    bool genericExpandedOnce_;
     QString strings_[STR__MAX];
     void onTabButtonChanged(int tab);
     void configClearButton();
-    void setListModel(QAbstractItemModel *model);
+    void setListModel(SessionTreeModel *model);
     void keyPressEvent(QKeyEvent * e);
 
   private slots:
     void on_pushButtonStart_clicked();
     void on_pushButtonAbort_clicked();
     void on_treeView_doubleClicked(const QModelIndex& index);
+    void on_treeView_expanded(const QModelIndex& index);
     void on_tabButtonLocal_clicked();
     void on_tabButtonMyClasses_clicked();
     void on_tabButtonAllClasses_clicked();