diff --git a/.npmrc b/.npmrc
new file mode 100644
index 00000000..3f675513
--- /dev/null
+++ b/.npmrc
@@ -0,0 +1,3 @@
+ignore-scripts=true
+//npm.pkg.github.com/:_authToken=<token>
+@codeflash-ai:registry=https://npm.pkg.github.com
diff --git a/package-lock.json b/package-lock.json
index 54982871..f3054534 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -12,7 +12,7 @@
         "@octokit/rest": "^20.1.2",
         "@types/yargs": "^16.0.0",
         "async-retry": "^1.3.1",
-        "diff": "^5.0.0",
+        "diff": "^7.0.0",
         "glob": "^7.1.6",
         "parse-diff": "^0.11.0",
         "yargs": "^16.0.0"
@@ -1894,9 +1894,10 @@
       "integrity": "sha512-xmHIy4F3scKVwMsQ4WnVaS8bHOx0DmVwRywosKhaILI0ywMDWPtBSku2HNxRvF7jtwDRsoEwYQSfbxj8b7RlJQ=="
     },
     "node_modules/diff": {
-      "version": "5.2.0",
-      "resolved": "https://registry.npmjs.org/diff/-/diff-5.2.0.tgz",
-      "integrity": "sha512-uIFDxqpRZGZ6ThOk84hEfqWoHx2devRFvpTZcTHur85vImfaxUbTW9Ryh4CpCuDnToOP1CEtXKIgytHBPVff5A==",
+      "version": "7.0.0",
+      "resolved": "https://registry.npmjs.org/diff/-/diff-7.0.0.tgz",
+      "integrity": "sha512-PJWHUb1RFevKCwaFA9RlG5tCd+FO5iRh9A8HEtkmBH2Li03iJriB6m6JIN4rGz3K3JLawI7/veA1xzRKP6ISBw==",
+      "license": "BSD-3-Clause",
       "engines": {
         "node": ">=0.3.1"
       }
@@ -4378,6 +4379,16 @@
         "node": ">=12"
       }
     },
+    "node_modules/mocha/node_modules/diff": {
+      "version": "5.2.0",
+      "resolved": "https://registry.npmjs.org/diff/-/diff-5.2.0.tgz",
+      "integrity": "sha512-uIFDxqpRZGZ6ThOk84hEfqWoHx2devRFvpTZcTHur85vImfaxUbTW9Ryh4CpCuDnToOP1CEtXKIgytHBPVff5A==",
+      "dev": true,
+      "license": "BSD-3-Clause",
+      "engines": {
+        "node": ">=0.3.1"
+      }
+    },
     "node_modules/mocha/node_modules/foreground-child": {
       "version": "3.3.1",
       "resolved": "https://registry.npmjs.org/foreground-child/-/foreground-child-3.3.1.tgz",
@@ -5583,6 +5594,16 @@
         "url": "https://opencollective.com/sinon"
       }
     },
+    "node_modules/sinon/node_modules/diff": {
+      "version": "5.2.0",
+      "resolved": "https://registry.npmjs.org/diff/-/diff-5.2.0.tgz",
+      "integrity": "sha512-uIFDxqpRZGZ6ThOk84hEfqWoHx2devRFvpTZcTHur85vImfaxUbTW9Ryh4CpCuDnToOP1CEtXKIgytHBPVff5A==",
+      "dev": true,
+      "license": "BSD-3-Clause",
+      "engines": {
+        "node": ">=0.3.1"
+      }
+    },
     "node_modules/slash": {
       "version": "3.0.0",
       "resolved": "https://registry.npmjs.org/slash/-/slash-3.0.0.tgz",
diff --git a/package.json b/package.json
index cd58c6e8..49a36625 100644
--- a/package.json
+++ b/package.json
@@ -1,16 +1,22 @@
 {
-  "name": "code-suggester",
+  "name": "@codeflash-ai/code-suggester",
   "description": "Library to propose code changes",
-  "version": "5.0.0",
+  "version": "5.0.2",
   "license": "Apache-2.0",
   "author": "Google LLC",
   "engines": {
     "node": ">=18.0.0"
   },
+  "publishConfig": {
+    "registry": "https://npm.pkg.github.com"
+  },
   "bin": {
     "code-suggester": "./build/src/bin/code-suggester.js"
   },
-  "repository": "googleapis/code-suggester",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/codeflash-ai/codeflash.git"
+  },
   "main": "build/src/index.js",
   "module": "build/src/index.js",
   "source": "build/src/index.js",
@@ -42,7 +48,7 @@
     "@octokit/rest": "^20.1.2",
     "@types/yargs": "^16.0.0",
     "async-retry": "^1.3.1",
-    "diff": "^5.0.0",
+    "diff": "^7.0.0",
     "glob": "^7.1.6",
     "parse-diff": "^0.11.0",
     "yargs": "^16.0.0"