diff --git a/AlgebraicDataflowArchitectureModel/src/application/editor/stages/ControlFlowModelingStage.java b/AlgebraicDataflowArchitectureModel/src/application/editor/stages/ControlFlowModelingStage.java
index 106c511..ae7d4f9 100644
--- a/AlgebraicDataflowArchitectureModel/src/application/editor/stages/ControlFlowModelingStage.java
+++ b/AlgebraicDataflowArchitectureModel/src/application/editor/stages/ControlFlowModelingStage.java
@@ -101,24 +101,24 @@
 	 * for resource nodes and event-channel nodes, and connects them with edges corresponding to the control flow.
 	 * Updates to the graph are performed within a single transaction for consistency.
 	 *
-	 * @param graph         The {@link mxGraph} instance to be constructed and updated.
-	 * @param resNodeToCell A map associating {@link ResourceNode} instances to their corresponding {@link mxCell}
-	 *                      instances in the graph.
-	 * @param channelToCell A map associating {@link DataTransferChannel} instances to their corresponding
-	 *                      {@link mxCell} instances in the graph.
+	 * @param graph                     The {@link mxGraph} instance to be constructed and updated.
+	 * @param originalResourceNodeCells A map associating {@link ResourceNode} instances to their corresponding {@link mxCell}
+	 *                                  instances in the graph.
+	 * @param originalChannelCells      A map associating {@link DataTransferChannel} instances to their corresponding
+	 *                                  {@link mxCell} instances in the graph.
 	 * @return The updated {@link mxGraph} instance after constructing the control flow graph.
 	 */
-	private mxGraph constructGraph(mxGraph graph, final Map<ResourceNode, mxCell> resNodeToCell, final Map<DataTransferChannel, mxCell> channelToCell) {
+	private mxGraph constructGraph(mxGraph graph, final Map<ResourceNode, mxCell> originalResourceNodeCells, final Map<DataTransferChannel, mxCell> originalChannelCells) {
 		showLayers(PUSH_FLOW_LAYER, PULL_FLOW_LAYER);
 		
 		graph.getModel().beginUpdate();
 		try {
 			// Create cells correspond to resource nodes for the PUSH/PULL layer.
-			Map<StatefulObjectNode, mxCell> pushResourceNodeCells = createResourceNodeCells(graph, resNodeToCell, PUSH_FLOW_LAYER);
-			Map<StatefulObjectNode, mxCell> pullResourceNodeCells = createResourceNodeCells(graph, resNodeToCell, PULL_FLOW_LAYER);
+			Map<StatefulObjectNode, mxCell> pushResourceNodeCells = createResourceNodeCells(graph, originalResourceNodeCells, PUSH_FLOW_LAYER);
+			Map<StatefulObjectNode, mxCell> pullResourceNodeCells = createResourceNodeCells(graph, originalResourceNodeCells, PULL_FLOW_LAYER);
 			
 			// Create cells correspond to event-channel nodes for the PUSH layer.
-			Map<EventChannelObjectNode, mxCell> pushFlowEventNodeCells = createEventChannelNodeCells(graph, channelToCell, PUSH_FLOW_LAYER);
+			Map<EventChannelObjectNode, mxCell> pushFlowEventNodeCells = createEventChannelNodeCells(graph, originalChannelCells, PUSH_FLOW_LAYER);
 			
 			// Insert edges between the connected vertices
 			graph = insertControlFlowEdges(PUSH_FLOW_LAYER, pushResourceNodeCells, pushFlowEventNodeCells);