1 files changed, 5 insertions, 2 deletions
diff --git a/src/mainwindow.h b/src/mainwindow.h
index 1376a84..c9a1844 100644
--- a/src/mainwindow.h
+++ b/src/mainwindow.h
@@ -28,14 +28,15 @@
 #ifndef __KAIMAN_MAINWINDOW_H__
 #define __KAIMAN_MAINWINDOW_H__
 
-#include <QWidget>
+#include <QMainWindow>
 
 #include <QFileSystemWatcher>
 
+#include "codeeditor.h"
 #include "outputwindow.h"
 #include "luascript.h"
 
-class MainWindow : public QWidget {
+class MainWindow : public QMainWindow {
 Q_OBJECT
 public:
   MainWindow(QString program);
@@ -44,6 +45,8 @@ private slots:
   void reset();
 
 private:
+  CodeEditor *editor;
+
   QFileSystemWatcher watcher;
   QString program;
   OutputWindow *out;