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();
diff --git a/AlgebraicDataflowArchitectureModel/src/generators/CodeGeneratorFromDataFlowGraph.java b/AlgebraicDataflowArchitectureModel/src/generators/CodeGeneratorFromDataFlowGraph.java
index a4b6fa6..875a141 100644
--- a/AlgebraicDataflowArchitectureModel/src/generators/CodeGeneratorFromDataFlowGraph.java
+++ b/AlgebraicDataflowArchitectureModel/src/generators/CodeGeneratorFromDataFlowGraph.java
@@ -50,6 +50,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.ResolvingMultipleDefinitionIsFutureWork;
@@ -60,7 +61,7 @@
 
 public class CodeGeneratorFromDataFlowGraph extends CodeGenerator {
 
-	public void generateCodeFromFlowGraph(DataTransferModel model, DataFlowGraph flowGraph, Collection<ResourceNode> components, ArrayList<CompilationUnit> codes, 
+	public void generateCodeFromFlowGraph(DataTransferModel model, IFlowGraph flowGraph, Collection<ResourceNode> components, ArrayList<CompilationUnit> codes, 
 			Map<ResourceHierarchy, Set<ResourceHierarchy>> dependedRootComponentGraph, IPlatformSpecific platformSpec, ILanguageSpecific langSpec) {
 		Map<ResourceHierarchy, TypeDeclaration> resourceComponents = new HashMap<>();
 		Map<ResourceHierarchy, MethodDeclaration> resourceConstructors = new HashMap<>();
diff --git a/AlgebraicDataflowArchitectureModel/src/models/dataFlowModel/DataFlowGraph.java b/AlgebraicDataflowArchitectureModel/src/models/dataFlowModel/DataFlowGraph.java
index a86dc97..3edaace 100644
--- a/AlgebraicDataflowArchitectureModel/src/models/dataFlowModel/DataFlowGraph.java
+++ b/AlgebraicDataflowArchitectureModel/src/models/dataFlowModel/DataFlowGraph.java
@@ -11,7 +11,7 @@
 import models.dataConstraintModel.ResourceHierarchy;
 import models.dataConstraintModel.ResourcePath;
 
-public class DataFlowGraph extends DirectedGraph {
+public class DataFlowGraph extends DirectedGraph implements IFlowGraph {
 	protected Set<ResourceNode> rootResourceNodes = null;
 	protected Set<ChannelNode> rootChannelNodes = null;
 	protected Map<DataTransferChannel, ChannelNode> channelNodeMap = null;
diff --git a/AlgebraicDataflowArchitectureModel/src/models/dataFlowModel/IFlowGraph.java b/AlgebraicDataflowArchitectureModel/src/models/dataFlowModel/IFlowGraph.java
new file mode 100644
index 0000000..40b107f
--- /dev/null
+++ b/AlgebraicDataflowArchitectureModel/src/models/dataFlowModel/IFlowGraph.java
@@ -0,0 +1,9 @@
+package models.dataFlowModel;
+
+import java.util.Collection;
+import java.util.Set;
+
+public interface IFlowGraph {
+	public Collection<ResourceNode> getResourceNodes();
+	public Set<ResourceNode> getRootResourceNodes();
+}