diff --git a/CMakeLists.txt b/CMakeLists.txt
index 8c11610..81361bb 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -123,10 +123,10 @@ set(CMAKE_INCLUDE_CURRENT_DIR ON)
 set(SOURCES
     src/main.cpp
     src/mainwindow.cpp
-    src/mainwindow.hpp
+    include/Air_Ground_CEC/mainwindow.hpp
     src/mainwindow.ui
     src/qnode.cpp
-    src/qnode.hpp
+    include/Air_Ground_CEC/qnode.hpp
 )
 ## Specify additional locations of header files
 ## Your package locations should be listed before other locations
diff --git a/include/Air_Ground_CEC/mainwindow.hpp b/include/Air_Ground_CEC/mainwindow.hpp
index 3671e71..d67fc60 100644
--- a/include/Air_Ground_CEC/mainwindow.hpp
+++ b/include/Air_Ground_CEC/mainwindow.hpp
@@ -2,8 +2,6 @@
 #define MAINWINDOW_H
 
 #include <QMainWindow>
-#include "ui_mainwindow.h"
-//#include "qnode.hpp"
 #include <QImage>
 #include <QMutex>
 
diff --git a/src/mainwindow.cpp b/src/mainwindow.cpp
index 443597b..5dc816b 100644
--- a/src/mainwindow.cpp
+++ b/src/mainwindow.cpp
@@ -1,4 +1,4 @@
-#include "mainwindow.hpp"
+#include "../include/Air_Ground_CEC/mainwindow.hpp"
 #include "ui_mainwindow.h"
 
 MainWindow::MainWindow(QWidget *parent) :