diff --git a/worker-model-generator/modules/worker-model-generator-core/build.gradle b/worker-model-generator/modules/worker-model-generator-core/build.gradle
index ec144a0d3..00a3140ca 100644
--- a/worker-model-generator/modules/worker-model-generator-core/build.gradle
+++ b/worker-model-generator/modules/worker-model-generator-core/build.gradle
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/AnnotationFinder.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/AnnotationFinder.java
index 4ee20331b..e1eaf2a2c 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/AnnotationFinder.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/AnnotationFinder.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/CommonUtils.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/CommonUtils.java
index 99d85c319..6512b072b 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/CommonUtils.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/CommonUtils.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/Constants.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/Constants.java
index bfdd47b8b..b0bf70e29 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/Constants.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/Constants.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/DocumentSymbolFinder.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/DocumentSymbolFinder.java
index ca155f913..368841f62 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/DocumentSymbolFinder.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/DocumentSymbolFinder.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/FlowBuilder.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/FlowBuilder.java
index 520448e69..79f324c8a 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/FlowBuilder.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/FlowBuilder.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
@@ -19,12 +19,14 @@
 package io.ballerina.workermodelgenerator.core;
 
 import io.ballerina.compiler.api.SemanticModel;
+import io.ballerina.compiler.syntax.tree.FunctionBodyBlockNode;
 import io.ballerina.compiler.syntax.tree.FunctionBodyNode;
 import io.ballerina.compiler.syntax.tree.FunctionDefinitionNode;
 import io.ballerina.compiler.syntax.tree.MetadataNode;
 import io.ballerina.compiler.syntax.tree.ModulePartNode;
 import io.ballerina.compiler.syntax.tree.NamedWorkerDeclarationNode;
 import io.ballerina.compiler.syntax.tree.NodeVisitor;
+import io.ballerina.compiler.syntax.tree.SyntaxKind;
 import io.ballerina.workermodelgenerator.core.analyzer.Analyzer;
 import io.ballerina.workermodelgenerator.core.model.CodeLocation;
 import io.ballerina.workermodelgenerator.core.model.Endpoint;
@@ -108,8 +110,14 @@ public void visit(NamedWorkerDeclarationNode namedWorkerDeclarationNode) {
     public void visit(FunctionDefinitionNode functionDefinitionNode) {
         // Obtain the code body location of the flow
         FunctionBodyNode functionBodyNode = functionDefinitionNode.functionBody();
-        setBodyCodeLocation(new CodeLocation(functionBodyNode.lineRange().startLine(),
-                functionBodyNode.lineRange().endLine()));
+
+        // TODO: Handle invalid function definitions
+        if (functionBodyNode.kind() != SyntaxKind.FUNCTION_BODY_BLOCK) {
+            return;
+        }
+        FunctionBodyBlockNode functionBodyBlockNode = (FunctionBodyBlockNode) functionBodyNode;
+        setBodyCodeLocation(new CodeLocation(functionBodyBlockNode.openBraceToken().lineRange().endLine(),
+                functionBodyBlockNode.closeBraceToken().lineRange().startLine()));
 
         // Process the flow metadata information from the annotations
         Optional<MetadataNode> metadata = functionDefinitionNode.metadata();
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/ModelGenerator.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/ModelGenerator.java
index 167054ba8..2879613de 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/ModelGenerator.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/ModelGenerator.java
@@ -1,3 +1,21 @@
+/*
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *
+ *  WSO2 LLC. licenses this file to you under the Apache License,
+ *  Version 2.0 (the "License"); you may not use this file except
+ *  in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing,
+ *  software distributed under the License is distributed on an
+ *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ *  KIND, either express or implied.  See the License for the
+ *  specific language governing permissions and limitations
+ *  under the License.
+ */
+
 package io.ballerina.workermodelgenerator.core;
 
 import com.google.gson.Gson;
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/NodeBuilder.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/NodeBuilder.java
index ea04999c4..4d287398d 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/NodeBuilder.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/NodeBuilder.java
@@ -1,3 +1,21 @@
+/*
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *
+ *  WSO2 LLC. licenses this file to you under the Apache License,
+ *  Version 2.0 (the "License"); you may not use this file except
+ *  in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing,
+ *  software distributed under the License is distributed on an
+ *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ *  KIND, either express or implied.  See the License for the
+ *  specific language governing permissions and limitations
+ *  under the License.
+ */
+
 package io.ballerina.workermodelgenerator.core;
 
 import io.ballerina.tools.text.LinePosition;
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/Analyzer.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/Analyzer.java
index 1113fcf9f..4fc0b95cc 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/Analyzer.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/Analyzer.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/CodeBlockAnalyzer.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/CodeBlockAnalyzer.java
index 83743d209..23e72fae8 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/CodeBlockAnalyzer.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/CodeBlockAnalyzer.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/HttpRequestAnalyzer.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/HttpRequestAnalyzer.java
index 1da7dbe59..070927faf 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/HttpRequestAnalyzer.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/HttpRequestAnalyzer.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/SwitchAnalyzer.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/SwitchAnalyzer.java
index ed664473f..55a72fa19 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/SwitchAnalyzer.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/SwitchAnalyzer.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/TransformAnalyzer.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/TransformAnalyzer.java
index 2bd5f7ab9..f94655653 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/TransformAnalyzer.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/analyzer/TransformAnalyzer.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/CanvasPosition.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/CanvasPosition.java
index b5e758f8a..aff3b68c0 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/CanvasPosition.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/CanvasPosition.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/CodeLocation.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/CodeLocation.java
index 7f736e79d..caf78f8af 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/CodeLocation.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/CodeLocation.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/Endpoint.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/Endpoint.java
index c316526f3..e9a6c00a4 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/Endpoint.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/Endpoint.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/Flow.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/Flow.java
index 15ba7de44..bd34ffc26 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/Flow.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/Flow.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/FlowJsonBuilder.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/FlowJsonBuilder.java
index 360a80a23..23cfe6384 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/FlowJsonBuilder.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/FlowJsonBuilder.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/InputPort.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/InputPort.java
index d7227345e..2eefd6fc5 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/InputPort.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/InputPort.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/OutputPort.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/OutputPort.java
index b8d7d7c13..c33331d07 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/OutputPort.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/OutputPort.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/WorkerNode.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/WorkerNode.java
index c3cf467cd..fef242ed5 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/WorkerNode.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/WorkerNode.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/WorkerNodeJsonBuilder.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/WorkerNodeJsonBuilder.java
index 9c7bfc047..48cb8fd4f 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/WorkerNodeJsonBuilder.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/WorkerNodeJsonBuilder.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/properties/BalExpression.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/properties/BalExpression.java
index 53b2ef862..b78136c3d 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/properties/BalExpression.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/properties/BalExpression.java
@@ -1,3 +1,21 @@
+/*
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *
+ *  WSO2 LLC. licenses this file to you under the Apache License,
+ *  Version 2.0 (the "License"); you may not use this file except
+ *  in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing,
+ *  software distributed under the License is distributed on an
+ *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ *  KIND, either express or implied.  See the License for the
+ *  specific language governing permissions and limitations
+ *  under the License.
+ */
+
 package io.ballerina.workermodelgenerator.core.model.properties;
 
 import io.ballerina.workermodelgenerator.core.model.CodeLocation;
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/properties/NodeProperties.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/properties/NodeProperties.java
index 99327b091..a1f203229 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/properties/NodeProperties.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/properties/NodeProperties.java
@@ -1,3 +1,21 @@
+/*
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *
+ *  WSO2 LLC. licenses this file to you under the Apache License,
+ *  Version 2.0 (the "License"); you may not use this file except
+ *  in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing,
+ *  software distributed under the License is distributed on an
+ *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ *  KIND, either express or implied.  See the License for the
+ *  specific language governing permissions and limitations
+ *  under the License.
+ */
+
 package io.ballerina.workermodelgenerator.core.model.properties;
 
 import io.ballerina.workermodelgenerator.core.model.CodeLocation;
@@ -6,6 +24,11 @@
 import java.util.List;
 import java.util.Objects;
 
+/**
+ * Represents the properties of a node.
+ *
+ * @since 2201.9.0
+ */
 public class NodeProperties {
 
     // Switch node properties
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/properties/SwitchCase.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/properties/SwitchCase.java
index 2edce5dc5..dd9949d62 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/properties/SwitchCase.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/properties/SwitchCase.java
@@ -1,8 +1,32 @@
+/*
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *
+ *  WSO2 LLC. licenses this file to you under the Apache License,
+ *  Version 2.0 (the "License"); you may not use this file except
+ *  in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing,
+ *  software distributed under the License is distributed on an
+ *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ *  KIND, either express or implied.  See the License for the
+ *  specific language governing permissions and limitations
+ *  under the License.
+ */
+
 package io.ballerina.workermodelgenerator.core.model.properties;
 
 import java.util.List;
 
-
+/**
+ * Represents a switch case.
+ *
+ * @param expression Expression of the case
+ * @param nodes      Output nodes relevant to the case
+ * @since 2201.9.0
+ */
 public record SwitchCase(BalExpression expression, List<String> nodes) {
 
 }
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/properties/SwitchDefaultCase.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/properties/SwitchDefaultCase.java
index 9efde3db5..22cfb62d1 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/properties/SwitchDefaultCase.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/io.ballerina.workermodelgenerator.core/model/properties/SwitchDefaultCase.java
@@ -1,7 +1,31 @@
+/*
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *
+ *  WSO2 LLC. licenses this file to you under the Apache License,
+ *  Version 2.0 (the "License"); you may not use this file except
+ *  in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing,
+ *  software distributed under the License is distributed on an
+ *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ *  KIND, either express or implied.  See the License for the
+ *  specific language governing permissions and limitations
+ *  under the License.
+ */
+
 package io.ballerina.workermodelgenerator.core.model.properties;
 
 import java.util.List;
 
+/**
+ * Represents a switch default case.
+ *
+ * @param nodes Output nodes relevant to the default case
+ * @since 2201.9.0
+ */
 public record SwitchDefaultCase(List<String> nodes) {
 
 }
diff --git a/worker-model-generator/modules/worker-model-generator-core/src/main/java/module-info.java b/worker-model-generator/modules/worker-model-generator-core/src/main/java/module-info.java
index e65d95f0d..f2b856779 100644
--- a/worker-model-generator/modules/worker-model-generator-core/src/main/java/module-info.java
+++ b/worker-model-generator/modules/worker-model-generator-core/src/main/java/module-info.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/build.gradle b/worker-model-generator/modules/worker-model-generator-ls-extension/build.gradle
index 7d3c98a62..429bf16e6 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/build.gradle
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/build.gradle
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorClientCapabilities.java b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorClientCapabilities.java
index 91d7c0884..a11b22bab 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorClientCapabilities.java
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorClientCapabilities.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorClientCapabilitySetter.java b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorClientCapabilitySetter.java
index af0998a3c..8ea8e2c92 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorClientCapabilitySetter.java
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorClientCapabilitySetter.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorConstants.java b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorConstants.java
index fffb3ab45..ac17aa168 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorConstants.java
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorConstants.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorServerCapabilities.java b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorServerCapabilities.java
index d46daf17c..5f57b11db 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorServerCapabilities.java
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorServerCapabilities.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorServerCapabilitySetter.java b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorServerCapabilitySetter.java
index 1d477f576..32cf01141 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorServerCapabilitySetter.java
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorServerCapabilitySetter.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorService.java b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorService.java
index 355953a71..1581aa47a 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorService.java
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorService.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorServiceRequest.java b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorServiceRequest.java
index b3684c853..a6b151cf6 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorServiceRequest.java
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorServiceRequest.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorServiceResponse.java b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorServiceResponse.java
index b5380fb67..48c28ab1b 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorServiceResponse.java
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/io/ballerina/workermodelgenerator/extension/WorkerModelGeneratorServiceResponse.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/module-info.java b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/module-info.java
index 7af252519..d94e15227 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/module-info.java
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/main/java/module-info.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2023, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
+ *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) All Rights Reserved.
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/alternate_receiver.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/alternate_receiver.json
index 0989529f2..8725942c5 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/alternate_receiver.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/alternate_receiver.json
@@ -16,11 +16,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 6,
-        "offset": 23
+        "offset": 24
       },
       "end": {
         "line": 43,
-        "offset": 1
+        "offset": 0
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/body_code_location.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/body_code_location.json
index 81b8c8f35..ac841b0c6 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/body_code_location.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/body_code_location.json
@@ -14,11 +14,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 10,
-        "offset": 75
+        "offset": 76
       },
       "end": {
         "line": 20,
-        "offset": 5
+        "offset": 4
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/code_block.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/code_block.json
index b5f907281..8e0a2e3d1 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/code_block.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/code_block.json
@@ -16,11 +16,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 6,
-        "offset": 23
+        "offset": 24
       },
       "end": {
         "line": 32,
-        "offset": 1
+        "offset": 0
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/hospital_service.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/hospital_service.json
index 00dd955ad..738b763b1 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/hospital_service.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/hospital_service.json
@@ -14,11 +14,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 12,
-        "offset": 75
+        "offset": 76
       },
       "end": {
         "line": 92,
-        "offset": 5
+        "offset": 4
       }
     },
     "fileSourceRange": {
@@ -108,7 +108,7 @@
       },
       {
         "name": "buildPineValleyPayload",
-        "templateId": "CloneNode",
+        "templateId": "NewPayloadNode",
         "codeLocation": {
           "start": {
             "line": 29,
@@ -137,7 +137,22 @@
             "name": "payload",
             "receiver": "callPineValley"
           }
-        ]
+        ],
+        "properties": {
+          "codeBlock": {
+            "expression": "            PineValleyPayload payload = {doctorType: doctorType};\n",
+            "codeLocation": {
+              "start": {
+                "line": 37,
+                "offset": 12
+              },
+              "end": {
+                "line": 37,
+                "offset": 65
+              }
+            }
+          }
+        }
       },
       {
         "name": "callPineValley",
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/http_empty.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/http_empty.json
index faeb20b68..324b359ae 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/http_empty.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/http_empty.json
@@ -16,11 +16,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 10,
-        "offset": 49
+        "offset": 50
       },
       "end": {
         "line": 25,
-        "offset": 1
+        "offset": 0
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/metadata_code_block.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/metadata_code_block.json
index 852cafc93..19e150964 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/metadata_code_block.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/metadata_code_block.json
@@ -16,11 +16,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 6,
-        "offset": 23
+        "offset": 24
       },
       "end": {
         "line": 16,
-        "offset": 1
+        "offset": 0
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/multi_empty_switch.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/multi_empty_switch.json
index 34a62886e..f3a0214a9 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/multi_empty_switch.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/multi_empty_switch.json
@@ -16,11 +16,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 6,
-        "offset": 23
+        "offset": 24
       },
       "end": {
         "line": 36,
-        "offset": 1
+        "offset": 0
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/multi_switch.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/multi_switch.json
index bf411538c..8bf822c29 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/multi_switch.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/multi_switch.json
@@ -16,11 +16,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 6,
-        "offset": 23
+        "offset": 24
       },
       "end": {
         "line": 86,
-        "offset": 1
+        "offset": 0
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/new_switch.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/new_switch.json
index af20065ad..64f43b26c 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/new_switch.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/new_switch.json
@@ -16,11 +16,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 6,
-        "offset": 38
+        "offset": 39
       },
       "end": {
         "line": 22,
-        "offset": 2
+        "offset": 0
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/output_send.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/output_send.json
index 8ab4ed692..f293a7d8f 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/output_send.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/output_send.json
@@ -16,11 +16,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 6,
-        "offset": 23
+        "offset": 24
       },
       "end": {
         "line": 43,
-        "offset": 1
+        "offset": 0
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/service_orchestration.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/service_orchestration.json
index 13fb2d511..54d3b546d 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/service_orchestration.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/service_orchestration.json
@@ -14,11 +14,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 83,
-        "offset": 128
+        "offset": 129
       },
       "end": {
         "line": 211,
-        "offset": 5
+        "offset": 4
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_flow.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_flow.json
index 0f10c09bd..64268122d 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_flow.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_flow.json
@@ -16,11 +16,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 6,
-        "offset": 23
+        "offset": 24
       },
       "end": {
         "line": 48,
-        "offset": 1
+        "offset": 0
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_http.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_http.json
index e557c39a4..65a2b4020 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_http.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_http.json
@@ -16,11 +16,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 10,
-        "offset": 49
+        "offset": 50
       },
       "end": {
         "line": 25,
-        "offset": 1
+        "offset": 0
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_import.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_import.json
index d3cc81a44..6f4ac5242 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_import.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_import.json
@@ -16,11 +16,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 7,
-        "offset": 36
+        "offset": 37
       },
       "end": {
         "line": 50,
-        "offset": 1
+        "offset": 0
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_project.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_project.json
index f52febeb2..b6ae24e07 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_project.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_project.json
@@ -16,11 +16,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 6,
-        "offset": 23
+        "offset": 24
       },
       "end": {
         "line": 63,
-        "offset": 1
+        "offset": 0
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_switch.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_switch.json
index e9620e365..b05653388 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_switch.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_switch.json
@@ -16,11 +16,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 6,
-        "offset": 23
+        "offset": 24
       },
       "end": {
         "line": 58,
-        "offset": 1
+        "offset": 0
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_transform.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_transform.json
index abab7f5d0..556a4ae79 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_transform.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/simple_transform.json
@@ -16,11 +16,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 14,
-        "offset": 23
+        "offset": 24
       },
       "end": {
         "line": 30,
-        "offset": 1
+        "offset": 0
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/transform_with_error.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/transform_with_error.json
index 25df16457..4918a7091 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/transform_with_error.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/transform_with_error.json
@@ -16,11 +16,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 14,
-        "offset": 23
+        "offset": 24
       },
       "end": {
         "line": 43,
-        "offset": 1
+        "offset": 0
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/transform_with_input.json b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/transform_with_input.json
index 08893a2ea..42cb4baa8 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/transform_with_input.json
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/config/transform_with_input.json
@@ -16,11 +16,11 @@
     "bodyCodeLocation": {
       "start": {
         "line": 14,
-        "offset": 23
+        "offset": 24
       },
       "end": {
         "line": 43,
-        "offset": 1
+        "offset": 0
       }
     },
     "fileSourceRange": {
diff --git a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/testng.xml b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/testng.xml
index 8dfbfc727..b0bfad426 100644
--- a/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/testng.xml
+++ b/worker-model-generator/modules/worker-model-generator-ls-extension/src/test/resources/testng.xml
@@ -1,6 +1,6 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <!--
-Copyright (c) 2023, WSO2 LLC. (http://www.wso2.org) All Rights Reserved.
+Copyright (c) 2024, WSO2 LLC. (http://www.wso2.org) All Rights Reserved.
 
 WSO2 LLC. licenses this file to you under the Apache License,
 Version 2.0 (the "License"); you may not use this file except