diff --git a/AlgebraicDataflowArchitectureModel/src/generators/CodeGeneratorFromDataFlowGraph.java b/AlgebraicDataflowArchitectureModel/src/generators/CodeGeneratorFromDataFlowGraph.java
index 19e3083..9a59356 100644
--- a/AlgebraicDataflowArchitectureModel/src/generators/CodeGeneratorFromDataFlowGraph.java
+++ b/AlgebraicDataflowArchitectureModel/src/generators/CodeGeneratorFromDataFlowGraph.java
@@ -15,7 +15,9 @@
 import code.ast.VariableDeclaration;
 import models.Edge;
 import models.Node;
+import models.algebra.Constant;
 import models.algebra.Expression;
+import models.algebra.Field;
 import models.algebra.InvalidMessage;
 import models.algebra.ParameterizedIdentifierIsFutureWork;
 import models.algebra.Term;
@@ -25,6 +27,7 @@
 import models.algebra.Variable;
 import models.dataConstraintModel.Channel;
 import models.dataConstraintModel.ChannelMember;
+import models.dataConstraintModel.DataConstraintModel;
 import models.dataConstraintModel.ResourcePath;
 import models.dataConstraintModel.Selector;
 import models.dataFlowModel.DataFlowEdge;
@@ -436,12 +439,12 @@
 					
 					if (input != null) {
 						// Add a statement to update the state field to the input method.
-						if (resourceNode.getNumberOfParameters() == 0) {
-							try {
-								String[] sideEffects = new String[] {""};
-								Expression updateExp;
-								updateExp = ((DataTransferChannel) ch).deriveUpdateExpressionOf(out, getPullAccessor());
-								String newState = updateExp.toImplementation(sideEffects);
+						try {
+							String[] sideEffects = new String[] {""};
+							Expression updateExp;
+							updateExp = ((DataTransferChannel) ch).deriveUpdateExpressionOf(out, getPullAccessor());
+							String newState = updateExp.toImplementation(sideEffects);
+							if (resourceNode.getResourceHierarchy().getNumParameters() == 0) {
 								String updateStatement;
 								if (updateExp instanceof Term && ((Term) updateExp).getSymbol().isImplWithSideEffect()) {
 									updateStatement = sideEffects[0];	
@@ -449,10 +452,37 @@
 									updateStatement = sideEffects[0] + langSpec.getFieldAccessor(fieldOfResourceState) + langSpec.getAssignment() + newState + langSpec.getStatementDelimiter();
 								}
 								input.addFirstStatement(updateStatement);
-							} catch (ParameterizedIdentifierIsFutureWork | ResolvingMultipleDefinitionIsFutureWork
-									| InvalidMessage | UnificationFailed | ValueUndefined e) {
-								e.printStackTrace();
+							} else {
+								String updateStatement = null;
+								if (updateExp instanceof Term && ((Term) updateExp).getSymbol().isImplWithSideEffect()) {
+									// ToDo.
+									updateStatement = sideEffects[0];	
+								} else {
+									if (DataConstraintModel.typeList.isAncestorOf(resourceNode.getParent().getResourceStateType())) {
+										Term selector = new Term(DataConstraintModel.set);
+										selector.addChild(new Constant(langSpec.getFieldAccessor(fieldOfResourceState)));
+										selector.addChild(new Variable(input.getParameters().get(input.getParameters().size() - 2).getName()));
+										selector.addChild(new Constant(newState));
+										String[] sideEffects2 = new String[] {""};
+										String newList = selector.toImplementation(sideEffects2);
+										updateStatement = sideEffects[0] + sideEffects2[0];
+									} else if (DataConstraintModel.typeMap.isAncestorOf(resourceNode.getParent().getResourceStateType())) {
+										Term selector = new Term(DataConstraintModel.set);
+										selector.addChild(new Constant(langSpec.getFieldAccessor(fieldOfResourceState)));
+										selector.addChild(new Variable(input.getParameters().get(input.getParameters().size() - 2).getName()));
+										selector.addChild(new Constant(newState));
+										String[] sideEffects2 = new String[] {""};
+										String newList = selector.toImplementation(sideEffects2);
+										updateStatement = sideEffects[0] + sideEffects2[0];
+									}
+									if (updateStatement != null) {
+										input.addFirstStatement(updateStatement);
+									}
+								}
 							}
+						} catch (ParameterizedIdentifierIsFutureWork | ResolvingMultipleDefinitionIsFutureWork
+								| InvalidMessage | UnificationFailed | ValueUndefined e) {
+							e.printStackTrace();
 						}
 												
 						// Add an invocation to an update method (for a chain of update method invocations).