diff --git a/AlgebraicDataflowArchitectureModel/src/application/editor/Editor.java b/AlgebraicDataflowArchitectureModel/src/application/editor/Editor.java
index da0e8bd..917370d 100644
--- a/AlgebraicDataflowArchitectureModel/src/application/editor/Editor.java
+++ b/AlgebraicDataflowArchitectureModel/src/application/editor/Editor.java
@@ -63,23 +63,6 @@
         stageChangeListeners = new ArrayList<>();
     }
 
-    public mxGraph getGraph() {
-        return graph;
-    }
-
-    public mxGraphComponent getGraphComponent() {
-        return graphComponent;
-    }
-
-    public DataTransferModel getModel() {
-        model = curStage.getModel();
-        return model;
-    }
-
-    public Stage getCurStage() {
-        return curStage;
-    }
-
     public boolean canChange(Stage nextStage) {
         return nextStage.canChangeFrom(curStage);
     }
@@ -131,22 +114,6 @@
         return null;
     }
 
-    public ArrayList<CompilationUnit> getCodes() {
-        return codes;
-    }
-
-    public void setCodes(ArrayList<CompilationUnit> codes) {
-        this.codes = codes;
-    }
-
-    public String getCurFileName() {
-        return curFileName;
-    }
-
-    public String getCurFilePath() {
-        return curFilePath;
-    }
-
     public void setCurFilePath(String curFilePath) {
         this.curFilePath = curFilePath;
         this.curFileName = new File(curFilePath).getName();
@@ -449,6 +416,39 @@
         return ((DataFlowModelingStage) curStage).connectEdge(edge, src, dst);
     }
 
+    public mxGraph getGraph() {
+        return graph;
+    }
+
+    public mxGraphComponent getGraphComponent() {
+        return graphComponent;
+    }
+
+    public DataTransferModel getModel() {
+        model = curStage.getModel();
+        return model;
+    }
+
+    public Stage getCurStage() {
+        return curStage;
+    }
+
+    public ArrayList<CompilationUnit> getCodes() {
+        return codes;
+    }
+
+    public void setCodes(ArrayList<CompilationUnit> codes) {
+        this.codes = codes;
+    }
+
+    public String getCurFileName() {
+        return curFileName;
+    }
+
+    public String getCurFilePath() {
+        return curFilePath;
+    }
+
     public static class SrcDstAttribute extends EdgeAttribute {
         private final Object src;
         private final Object dst;