diff --git a/demos/airplane_assembly_demo.py b/airplane_assembly_demo.py
similarity index 100%
rename from demos/airplane_assembly_demo.py
rename to airplane_assembly_demo.py
diff --git a/demos/q_values.p b/data/q_values.p
similarity index 100%
rename from demos/q_values.p
rename to data/q_values.p
diff --git a/demos/states.p b/data/states.p
similarity index 100%
rename from demos/states.p
rename to data/states.p
diff --git a/src/gui.py b/src/gui.py
index 3b2fcc9..fa27a92 100644
--- a/src/gui.py
+++ b/src/gui.py
@@ -18,6 +18,13 @@ def __init__(self):
         self.query.adjustSize()
         self.query.move(120, 50)
 
+        # task info
+        self.image = QLabel(self)
+        pixmap = QPixmap('task.jpg')
+        self.image.setPixmap(pixmap)
+        self.image.adjustSize()
+        self.image.move(450, 50)
+
         # inputs
         self.options = []
         self.user_choice = []
diff --git a/src/task.jpg b/src/task.jpg
index 720ff52..da0f0e8 100644
Binary files a/src/task.jpg and b/src/task.jpg differ