diff --git a/ai-coder.sh b/ai-coder.sh
index 8a56aa8..988e46e 100755
--- a/ai-coder.sh
+++ b/ai-coder.sh
@@ -43,7 +43,7 @@ if ! [ "$other_args" = "" ]; then
   if [ "$other_args" = "sleep" ]; then
     OLLAMA_HOST=$host ollama stop $model
   elif [ "$other_args" = "init" ]; then
-    ollama create "$model" -f "$modelfile"
+    OLLAMA_HOST=$host ollama create "$model" -f "$modelfile"
   else
     # If -t flag is set, use typora to display output
     if [ "$typora_flag" = true ]; then
diff --git a/ai.sh b/ai.sh
index ebab2cd..5fc93d9 100755
--- a/ai.sh
+++ b/ai.sh
@@ -48,6 +48,10 @@ if ! [ "$other_args" = "" ]; then
       deactivate
   elif [ "$other_args" = "remote" ]; then
       export OLLAMA_HOST=192.168.0.6
+  elif [ "$other_args" = "list" ]; then
+    OLLAMA_HOST=$host ollama list
+  elif [ "$other_args" = "ps" ]; then
+    OLLAMA_HOST=$host ollama ps
   elif [ "$other_args" = "init" ]; then
     OLLAMA_HOST=$host ollama create $model -f $modelfile
   elif [ "$other_args" = "wake" ]; then