diff --git a/Dockerfile.ollama b/Dockerfile.ollama
index 6d843fd..0fcd46d 100644
--- a/Dockerfile.ollama
+++ b/Dockerfile.ollama
@@ -1,7 +1,5 @@
 FROM ollama/ollama
 
-RUN ollama pull llama3.1:8b-instruct-q5_K_M
-
 COPY start.sh /start.sh
 RUN chmod +x /start.sh
 
diff --git a/data/config.json b/data/chatbot/config.json
similarity index 100%
rename from data/config.json
rename to data/chatbot/config.json
diff --git a/docker-compose.yaml b/docker-compose.yaml
index d68bc18..572a45e 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -1,12 +1,11 @@
 services:
 
   ollama:
-    hostname: ollama
-    container_name: ollama
+    image: ollama/ollama
     build:
       context: .
       dockerfile: Dockerfile.ollama
-    image: ollama/ollama
+    hostname: ollama
     container_name: ollama
     ports:
       - "11434:11434"
@@ -16,11 +15,12 @@ services:
     restart: unless-stopped
 
   matrix-chatbot:
-    container_name: matrix-chatbot
-    hostname: matrix-chatbot
+    image: matrix-chatbot:latest
     build:
       context: .
       dockerfile: Dockerfile.chatbot
+    container_name: matrix-chatbot
+    hostname: matrix-chatbot
     networks:
       dockernet:
         ipv4_address: 172.16.0.50