1 files changed, 4 insertions, 1 deletions
diff --git a/src/ui/systeminfostorageui.h b/src/ui/systeminfostorageui.h
index aab3741..734a890 100644
--- a/src/ui/systeminfostorageui.h
+++ b/src/ui/systeminfostorageui.h
@@ -37,6 +37,7 @@
 
 class QLineEdit;
 class QComboBox;
+class QPushButton;
 
 class StorageSystemInfoUi;
 
@@ -123,13 +124,15 @@ private slots:
 private:
     void initializeStorage();
     void initializeStorageOptions();
-
+    Q_INVOKABLE void enableDriveControls(bool enabled);
 
 private:
     QComboBox *systemInfoDrives;
     QComboBox *systemInfoDriveType;
     QLineEdit *systemInfoDriveTotalSpace;
     QLineEdit *systemInfoDriveAvailableSpace;
+    QPushButton *systemInfoRemoveDrive;
+    QPushButton *systemInfoChangeDriveName;
 
     StorageData mData;