diff --git a/AlgebraicDataflowArchitectureModel/src/generators/CodeGenerator.java b/AlgebraicDataflowArchitectureModel/src/generators/CodeGenerator.java
index b23a6a3..44cf225 100644
--- a/AlgebraicDataflowArchitectureModel/src/generators/CodeGenerator.java
+++ b/AlgebraicDataflowArchitectureModel/src/generators/CodeGenerator.java
@@ -38,6 +38,7 @@
 import models.dataFlowModel.DataFlowGraph;
 import models.dataFlowModel.DataTransferChannel;
 import models.dataFlowModel.DataTransferModel;
+import models.dataFlowModel.IFlowGraph;
 import models.dataFlowModel.PushPullAttribute;
 import models.dataFlowModel.PushPullValue;
 import models.dataFlowModel.ResourceNode;
@@ -137,7 +138,7 @@
 	 * @param langSpec specified language
 	 * @return source codes
 	 */
-	public ArrayList<CompilationUnit> generateCode(DataTransferModel model, DataFlowGraph flowGraph, IPlatformSpecific platformSpec, ILanguageSpecific langSpec) {
+	public ArrayList<CompilationUnit> generateCode(DataTransferModel model, IFlowGraph flowGraph, IPlatformSpecific platformSpec, ILanguageSpecific langSpec) {
 		CodeGenerator.langSpec = langSpec;
 		CodeGenerator.platformSpec = platformSpec;
 		ArrayList<CompilationUnit> codes = new ArrayList<>();
@@ -163,7 +164,7 @@
 		return codes;
 	}
 	
-	public abstract void generateCodeFromFlowGraph(DataTransferModel model, DataFlowGraph flowGraph, Collection<ResourceNode> components, ArrayList<CompilationUnit> codes, 
+	public abstract void generateCodeFromFlowGraph(DataTransferModel model, IFlowGraph flowGraph, Collection<ResourceNode> components, ArrayList<CompilationUnit> codes, 
 			Map<ResourceHierarchy, Set<ResourceHierarchy>> dependedRootComponentGraph, IPlatformSpecific platformSpec, ILanguageSpecific langSpec);
 	
 	private static Map<ResourceHierarchy, Set<ResourceHierarchy>> getDependedRootComponentGraph(DataTransferModel model) {
@@ -207,7 +208,7 @@
 		}
 	}
 	
-	private static ArrayList<ResourceNode> determineComponentOrder(DataFlowGraph graph, Map<ResourceHierarchy, Set<ResourceHierarchy>> dependedRootComponentGraph) {
+	private static ArrayList<ResourceNode> determineComponentOrder(IFlowGraph graph, Map<ResourceHierarchy, Set<ResourceHierarchy>> dependedRootComponentGraph) {
 		ArrayList<ResourceNode> objects = new ArrayList<>();
 		Set<ResourceNode> visited = new HashSet<>();
 		Collection<ResourceNode> allNodes = graph.getResourceNodes();