diff --git a/gui.py b/gui.py
index 276d852..e521731 100644
--- a/gui.py
+++ b/gui.py
@@ -47,6 +47,8 @@ layout = [
         sg.VSeparator(),
         sg.Column(file_list_column),
         sg.VSeparator(),
+        sg.Button(button_text='Create Image', key='-CREATE-'),
+        sg.VSeparator(),
         sg.Column(diagramm_viewer_column),
         
     ]
@@ -107,10 +109,22 @@ while True:
             )
 
             window["-TOUT-"].update(filename)
-            nassi(filename)
+            #nassi(filename)
 
         except:
             pass
+    elif event == '-CREATE-':
+        if values['-JAVA FOLDER-'] and values['-OUTPUT FOLDER-']:
+            filename = os.path.join(
+                values["-JAVA FOLDER-"], values["-JAVA FILE LIST-"][0]
+            )
+            nassi(filename)
+        elif values['-JAVA FILE LIST-']:
+            print('No Output')
+        elif values['-OUTPUT FOLDER-']:
+            print('No Input')
+        else:
+            pass
     
 
 window.close()