diff --git a/AlgebraicDataflowArchitectureModel/src/application/ApplicationWindow.java b/AlgebraicDataflowArchitectureModel/src/application/ApplicationWindow.java
index 8d20abc..4630830 100644
--- a/AlgebraicDataflowArchitectureModel/src/application/ApplicationWindow.java
+++ b/AlgebraicDataflowArchitectureModel/src/application/ApplicationWindow.java
@@ -19,7 +19,7 @@
 import application.editor.Editor;
 import application.editor.stages.DataFlowCellEditor;
 import application.views.NavigationWindow;
-import application.views.controlFlowDelegation.FlowLayerWindow;
+import application.views.controlFlowDelegation.ShowFlowWindow;
 
 public class ApplicationWindow extends JFrame {
 	private static final long serialVersionUID = -8690140317781055614L;
@@ -31,7 +31,7 @@
 
 	private ApplicationMenuBar menuBar = null;
 	private NavigationWindow navigationWindow = null;
-	private FlowLayerWindow switchDisplayFlowWindow = null;
+	private ShowFlowWindow FlowLayerWindow = null;
 
 	public ApplicationWindow() {
 		setTitle(title);
@@ -65,10 +65,11 @@
 		navigationWindow = new NavigationWindow(this, editor);
 		navigationWindow.setVisible(true);
 		
-		switchDisplayFlowWindow = new FlowLayerWindow(this);
-		switchDisplayFlowWindow.setVisible(false);
+		FlowLayerWindow = new ShowFlowWindow(this);
+		FlowLayerWindow.setVisible(false);
 		
 		editor.addStageChangeListener(navigationWindow);
+		editor.addStageChangeListener(FlowLayerWindow);
 	}
 
 	public mxGraph getGraph() {
@@ -92,7 +93,7 @@
 	}
 	
 	public void showSwitchLayerWindow() {
-		switchDisplayFlowWindow.setVisible(true);
+		FlowLayerWindow.setVisible(true);
 	}
 
 }