diff --git a/AlgebraicDataflowArchitectureModel/src/graphicalrefactor/actions/PrototypeGenerateAction.java b/AlgebraicDataflowArchitectureModel/src/graphicalrefactor/actions/PrototypeGenerateAction.java
index f5f090e..facc2cf 100644
--- a/AlgebraicDataflowArchitectureModel/src/graphicalrefactor/actions/PrototypeGenerateAction.java
+++ b/AlgebraicDataflowArchitectureModel/src/graphicalrefactor/actions/PrototypeGenerateAction.java
@@ -24,9 +24,7 @@
 		DataFlowModel model = editor.getModel();
 		DataStorageDecision.run(graph);
 		editor.setCodes(MethodBodyGenerator.doGenerate(graph, model, CodeGenerator.doGenerate(graph, model)));
-		for (TypeDeclaration type: editor.getCodes()) {
-			System.out.println(type);
-		}
+		System.out.println(editor.getCodes());
 	}
 
 }
diff --git a/AlgebraicDataflowArchitectureModel/src/tests/CodeGeneratorTest.java b/AlgebraicDataflowArchitectureModel/src/tests/CodeGeneratorTest.java
index b8cc590..4b652d8 100644
--- a/AlgebraicDataflowArchitectureModel/src/tests/CodeGeneratorTest.java
+++ b/AlgebraicDataflowArchitectureModel/src/tests/CodeGeneratorTest.java
@@ -34,9 +34,7 @@
 						System.out.println("-" + method.getReturnType().getTypeName() + " " + method.getName());
 					}
 				}
-				for(String str:CodeGenerator.getCodes(codetree)) {
-					System.out.println(str);
-				}
+				System.out.println(codetree);
 			} catch (ExpectedChannel | ExpectedChannelName | ExpectedLeftCurlyBracket | ExpectedInOrOutKeyword
 					| ExpectedStateTransition | ExpectedEquals | ExpectedRHSExpression | WrongLHSExpression
 					| WrongRHSExpression | ExpectedRightBracket e) {