diff --git a/AlgebraicDataflowArchitectureModel/src/algorithms/MethodBodyGenerator.java b/AlgebraicDataflowArchitectureModel/src/algorithms/MethodBodyGenerator.java
index 21a2454..c27b6b3 100644
--- a/AlgebraicDataflowArchitectureModel/src/algorithms/MethodBodyGenerator.java
+++ b/AlgebraicDataflowArchitectureModel/src/algorithms/MethodBodyGenerator.java
@@ -10,6 +10,7 @@
 import models.Node;
 import models.algebra.InvalidMessage;
 import models.algebra.ParameterizedIdentifierIsFutureWork;
+import models.algebra.Symbol;
 import models.algebra.UnificationFailed;
 import models.algebra.ValueUndefined;
 import models.dataConstraintModel.ChannelMember;
@@ -23,6 +24,21 @@
 
 public class MethodBodyGenerator {
 	public static ArrayList<TypeDeclaration> doGenerate(ResourceDependencyGraph graph, DataFlowModel model, ArrayList<TypeDeclaration> types) {
+		Symbol floor = model.getSymbol("floor");
+		Symbol.Memento floorMem = null;
+		if (floor != null) {
+			floorMem = floor.createMemento();
+			floor.setImplName("(int)Math.floor");
+			floor.setImplOperatorType(Symbol.Type.PREFIX);
+		}
+		Symbol sum = model.getSymbol("sum");
+		Symbol.Memento sumMem = null;
+		if (sum != null) {
+			sumMem = sum.createMemento();
+			sum.setImplName("stream().mapToInt(x->x).sum");
+			sum.setImplOperatorType(Symbol.Type.METHOD);
+		}
+
 		// Create a map from type names (lower case) to their types.
 		Map<String, TypeDeclaration> typeMap = new HashMap<>();
 		for (TypeDeclaration type: types) {
@@ -36,16 +52,22 @@
 				PushPullAttribute pushPull = (PushPullAttribute) d.getAttribute();
 				ResourceNode src = (ResourceNode) d.getSource();
 				ResourceNode dst = (ResourceNode) d.getDestination();
-				TypeDeclaration dstType = typeMap.get(dst.getIdentifierTemplate().getResourceName());
+				String srcResourceName = src.getIdentifierTemplate().getResourceName();
+				String dstResourceName = dst.getIdentifierTemplate().getResourceName();
+				TypeDeclaration srcType = typeMap.get(srcResourceName);
+				TypeDeclaration dstType = typeMap.get(dstResourceName);
 				for (ChannelMember out: d.getChannelGenerator().getOutputChannelMembers()) {
 					if (out.getIdentifierTemplate() == dst.getIdentifierTemplate()) {
 						if (pushPull.getOptions().get(0) == PushPullValue.PUSH) {
 							// push
 							MethodDeclaration update = getUpdateMethod(dstType);
 							String curState = d.getChannelGenerator().deriveUpdateExpressionOf(out, CodeGenerator.pushAccessor).toImplementation();
-							update.addStatement(dst.getIdentifierTemplate().getResourceName() + " = " + curState + ";");
+							update.addFirstStatement(dstResourceName + " = " + curState + ";");
 							MethodDeclaration getter = getGetterMethod(dstType);
-							getter.addStatement("return " + dst.getIdentifierTemplate().getResourceName() + ";");
+							getter.addStatement("return " + dstResourceName + ";");
+							// src side
+							MethodDeclaration srcUpdate = getUpdateMethod(srcType);
+							if (srcUpdate != null) srcUpdate.addStatement(dstResourceName + ".update(" + srcResourceName + ");");
 						} else {
 							// pull or push/pull
 							MethodDeclaration getter = getGetterMethod(dstType);
@@ -67,6 +89,9 @@
 				| InvalidMessage | UnificationFailed | ValueUndefined e1) {
 			e1.printStackTrace();
 		}
+		
+		if (floor != null) floor.setMemento(floorMem);
+		if (sum != null) sum.setMemento(sumMem);
 		return types;
 	}