diff --git a/AlgebraicDataflowArchitectureModel/src/algorithms/CodeGenerator.java b/AlgebraicDataflowArchitectureModel/src/algorithms/CodeGenerator.java
index ae1da6a..bbf1201 100644
--- a/AlgebraicDataflowArchitectureModel/src/algorithms/CodeGenerator.java
+++ b/AlgebraicDataflowArchitectureModel/src/algorithms/CodeGenerator.java
@@ -154,6 +154,11 @@
 					if (!method.getParameters().isEmpty())
 						varstr = varstr.substring(0, varstr.length() - 1);
 				}
+				if (method.getBody() != null) {
+					for (String str : method.getBody().getStatements()) {
+						codes.add("\t\t" + str + ";");
+					}
+				}
 				codes.add(varstr + ")" + "{");
 				codes.add("\t" + "}");
 				codes.add("");