diff --git a/.github/workflows/ai-review.yml b/.github/workflows/ai-review.yml
index 5f1678c..0df26ca 100644
--- a/.github/workflows/ai-review.yml
+++ b/.github/workflows/ai-review.yml
@@ -5,6 +5,7 @@ on:
     types: [opened, synchronize]
 
 permissions:
+  contents: read
   pull-requests: write
 
 jobs:
@@ -14,7 +15,7 @@ jobs:
       - name: Checkout
         uses: actions/checkout@v4
       - name: Code Review
-        uses: tarmojussila/zai-code-review@v1
+        uses: tarmojussila/zai-code-review@main
         with:
           ZAI_API_KEY: ${{ secrets.ZAI_API_KEY }}
           ZAI_MODEL: ${{ vars.ZAI_MODEL }}
diff --git a/README.md b/README.md
index 09fd4b4..808f121 100644
--- a/README.md
+++ b/README.md
@@ -25,7 +25,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
-      - uses: tarmojussila/zai-code-review@v1
+      - uses: tarmojussila/zai-code-review@main
         with:
           api_key: ${{ secrets.ZAI_API_KEY }}
 ```
diff --git a/test/sample.js b/test/sample.js
new file mode 100644
index 0000000..c8b8424
--- /dev/null
+++ b/test/sample.js
@@ -0,0 +1,17 @@
+function getUserData(userId) {
+  var data = [];
+  for (var i = 0; i < 1000; i++) {
+    data.push({ id: i, value: Math.random() });
+  }
+  return data.find(function(item) {
+    return item.id == userId;
+  });
+}
+
+function calculateTotal(items) {
+  var total = 0;
+  for (var i = 0; i <= items.length; i++) {
+    total = total + items[i].price;
+  }
+  return total;
+}