diff --git a/.vscode/extensions.json b/.vscode/extensions.json
index 8166507..c07aa91 100644
--- a/.vscode/extensions.json
+++ b/.vscode/extensions.json
@@ -1,5 +1,6 @@
 {
     "recommendations": [
+        "saoudrizwan.claude-dev",
         "nr-codetools.localaipilot",
         "continue.continue"
     ]
diff --git a/llama3.2/3b b/llama3.2/3b
index 574f741..d6fdf2b 100755
--- a/llama3.2/3b
+++ b/llama3.2/3b
@@ -1,4 +1,4 @@
 #!/bin/bash
 # https://ollama.com/library/llama3.2/tags
 
-docker exec -it ai-ollama ollama run llama3.2:1b --verbose
+docker exec -it ai-ollama ollama run llama3.2:3b --verbose
diff --git a/ollama.sh b/ollama.sh
index 7c47796..7eff5d3 100755
--- a/ollama.sh
+++ b/ollama.sh
@@ -1,3 +1,3 @@
 #!/bin/bash
 
-docker exec -it ai-ollama bash
+docker exec -it ai-ollama ollama "$@"
diff --git a/run.sh b/run.sh
index 2a99f84..03cbfee 100755
--- a/run.sh
+++ b/run.sh
@@ -2,19 +2,19 @@
 
 docker run \
     --pull=always \
+    --rm \
     --detach \
     --volume ./.data/ollama:/root/.ollama \
     --publish 11434:11434 \
-    --restart unless-stopped \
     --name ai-ollama \
     ollama/ollama
 
 docker run \
     --pull=always \
+    --rm \
     --detach \
     --volume ./.data/webui:/app/backend/data \
     --publish 9999:8080 \
-    --restart unless-stopped \
     --add-host=host.docker.internal:host-gateway \
     --name ai-webui \
     ghcr.io/open-webui/open-webui:main