diff --git a/AlgebraicDataflowArchitectureModel/src/generators/CodeGenerator.java b/AlgebraicDataflowArchitectureModel/src/generators/CodeGenerator.java
index f4a867b..3ce6bb8 100644
--- a/AlgebraicDataflowArchitectureModel/src/generators/CodeGenerator.java
+++ b/AlgebraicDataflowArchitectureModel/src/generators/CodeGenerator.java
@@ -506,6 +506,26 @@
 		}
 		return input;
 	}
+	
+
+	protected MethodDeclaration getMethod(TypeDeclaration component, String methodName, List<VariableDeclaration> params) {
+		for (MethodDeclaration m: component.getMethods()) {
+			if (m.getName().equals(methodName)) {
+				if (m.getParameters() == null && (params == null || params.size() == 0)) return m;
+				if (m.getParameters() != null && params != null && m.getParameters().size() == params.size()) {
+					boolean matchParams = true;
+					for (int i = 0; i < m.getParameters().size(); i++) {
+						if (!m.getParameters().get(i).getType().equals(params.get(i).getType())) {
+							matchParams = false;
+							break;
+						}
+					}
+					if (matchParams) return m;
+				}
+			}
+		}
+		return null;
+	}
 
 	protected MethodDeclaration getMethod(TypeDeclaration component, String methodName) {
 		for (MethodDeclaration m: component.getMethods()) {