1 files changed, 17 insertions, 5 deletions
diff --git a/vaporize/library/ControlFlowGraph.java b/vaporize/library/ControlFlowGraph.java
index f181d36..773c4cd 100644
--- a/vaporize/library/ControlFlowGraph.java
+++ b/vaporize/library/ControlFlowGraph.java
@@ -5,9 +5,9 @@ import java.util.ArrayList;
 
 public class ControlFlowGraph {
 
-    private ArrayList<Node> nodes;
-    private Node start;
-    private Node end;
+    private ArrayList<CFGNode> nodes;
+    private CFGNode start;
+    private CFGNode end;
 
     public ControlFlowGraph() {
         this.nodes = new ArrayList<>();
@@ -15,13 +15,25 @@ public class ControlFlowGraph {
         this.end = null;
     }
 
-    protected void addNode(Node node) {
+    protected void addNode(CFGNode node) {
         this.nodes.add(node);
     }
 
-    protected void addEdge(Node source, Node dest) {
+    protected void addEdge(CFGNode source, CFGNode dest) {
         source.addDest(dest);
         dest.addSource(source);
     }
 
+    protected void setStart(CFGNode start) {
+        this.start = start;
+    }
+
+    /**
+     * The following methods are for testing
+     * only!
+     */
+    protected ArrayList<CFGNode> getNodes() {
+        return this.nodes;
+    }
+
 }