diff --git a/app/elementstablemodel.h b/app/elementstablemodel.h
index 1b5147cb417dcfdc8e4cf970b79bfd0672942d11..bce3c13c2cb5238bee95674ea71b842e8f469e10 100644
--- a/app/elementstablemodel.h
+++ b/app/elementstablemodel.h
@@ -4,6 +4,7 @@
 #include <QAbstractTableModel>
 #include <task.h>
 #include <QDebug>
+#include <QFile>
 
 
 class ElementsTableModel : public QAbstractTableModel
@@ -35,6 +36,8 @@ public:
     // Remove data:
     bool removeRows(int row, int count, const QModelIndex &parent = QModelIndex()) override;
     bool removeColumns(int column, int count, const QModelIndex &parent = QModelIndex()) override;
+    
+    void writeCSV(QFile file);
 
 private: