diff --git a/vscode-extension/package.json b/vscode-extension/package.json
index 86a8dc101..6bedb382d 100644
--- a/vscode-extension/package.json
+++ b/vscode-extension/package.json
@@ -5,14 +5,43 @@
   "author": "LastMile AI",
   "displayName": "AIConfig Editor",
   "description": "AIConfig notebook editor that turns VSCode into a generative AI studio.",
-  "version": "0.0.18",
+  "version": "0.0.20",
   "license": "MIT",
   "engines": {
     "vscode": "^1.85.0"
   },
   "icon": "static/Logo.png",
   "categories": [
-    "Other"
+    "Machine Learning",
+    "Notebooks",
+    "Programming Languages",
+    "Snippets"
+  ],
+  "keywords": [
+    "ai",
+    "assistant",
+    "chatbot",
+    "notebook",
+    "aiconfig",
+    "lastmile",
+    "openai",
+    "chatgpt",
+    "gpt",
+    "gpt4",
+    "dalle",
+    "dalle3",
+    "llama",
+    "llama2",
+    "anthropic",
+    "claude",
+    "claude3",
+    "gemini",
+    "huggingface",
+    "transformers",
+    "generative",
+    "editor",
+    "python",
+    "typescript"
   ],
   "main": "./out/extension.js",
   "activationEvents": [
@@ -269,4 +298,4 @@
   "extensionDependencies": [
     "ms-python.python"
   ]
-}
\ No newline at end of file
+}