diff --git a/src/Edge.java b/src/Edge.java
index 067edd5..cee5ea2 100644
--- a/src/Edge.java
+++ b/src/Edge.java
@@ -1,6 +1,6 @@
 public class Edge {
-    Node source;
-    Node destination;
+    GraphNode source;
+    GraphNode destination;
     int weight;
 
     public Edge(Node source, Node destination, int weight) {
diff --git a/src/Node.java b/src/Node.java
index c9b2877..17e38c5 100644
--- a/src/Node.java
+++ b/src/Node.java
@@ -4,6 +4,7 @@ import java.util.List;
 interface GraphNode {
     String name = "";
     List<Edge> edges = new ArrayList<Edge>();
+    String getName();
 }
 
 public class Node implements GraphNode {