diff --git a/AlgebraicDataflowArchitectureModel/src/algorithms/JavaCodeGenerator.java b/AlgebraicDataflowArchitectureModel/src/algorithms/JavaCodeGenerator.java
index 93f04d5..8b4c569 100644
--- a/AlgebraicDataflowArchitectureModel/src/algorithms/JavaCodeGenerator.java
+++ b/AlgebraicDataflowArchitectureModel/src/algorithms/JavaCodeGenerator.java
@@ -2,6 +2,7 @@
 
 import java.util.ArrayList;
 import java.util.HashSet;
+import java.util.Set;
 
 import code.ast.Block;
 import code.ast.CompilationUnit;
@@ -57,13 +58,15 @@
 
 	static public ArrayList<CompilationUnit> doGenerate(ResourceDependencyGraph graph, DataFlowModel model) {
 		ArrayList<CompilationUnit> codes = new ArrayList<>();
-		ArrayList<ResourceNode> resources = StoreResourceCheck(graph);
+//		ArrayList<ResourceNode> resources = StoreResourceCheck(graph);
+		Set<Node> resources = graph.getNodes();
 
 		TypeDeclaration mainType = new TypeDeclaration(mainTypeName);
 		CompilationUnit mainCU = new CompilationUnit(mainType);
 		mainCU.addImport(new ImportDeclaration("java.util.*"));
 		codes.add(mainCU);
-		for (ResourceNode rn : resources) {
+		for (Node n : resources) {
+			ResourceNode rn = (ResourceNode) n;
 			boolean f = false;
 			String resourceName = rn.getIdentifierTemplate().getResourceName().substring(0, 1).toUpperCase()
 					+ rn.getIdentifierTemplate().getResourceName().substring(1);