diff --git a/AlgebraicDataflowArchitectureModel/src/graphicalrefactor/actions/PrototypeGenerateAction.java b/AlgebraicDataflowArchitectureModel/src/graphicalrefactor/actions/PrototypeGenerateAction.java
index 392dbc5..956f4b6 100644
--- a/AlgebraicDataflowArchitectureModel/src/graphicalrefactor/actions/PrototypeGenerateAction.java
+++ b/AlgebraicDataflowArchitectureModel/src/graphicalrefactor/actions/PrototypeGenerateAction.java
@@ -21,7 +21,10 @@
 	public void actionPerformed(ActionEvent e) {
 		ResourceDependencyGraph graph = editor.getResourceGraph();
 		DataStorageDecision.run(graph);
-		CodeGenerator.doGenerate(graph);
+		editor.setCodes(CodeGenerator.doGenerate(graph));
+		for(String str:editor.getCodes()) {
+			System.out.println(str);
+		}
 	}
 
 }