diff --git a/Dockerfile b/Dockerfile
index d09def6..e161f9c 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -32,7 +32,7 @@ RUN apt-get update --yes --quiet && DEBIAN_FRONTEND=noninteractive apt-get insta
 WORKDIR /work
 
 # Add my src as /work
-ADD engine.py handler.py requirements.txt start.sh test_input.json utils.py utils.py /work
+ADD ./src /work
 
 # Set defaut ollama models directory to /runpod-volume where runpod will mount the volume by default
 ENV OLLAMA_MODELS="/runpod-volume"
diff --git a/engine.py b/src/engine.py
similarity index 100%
rename from engine.py
rename to src/engine.py
diff --git a/handler.py b/src/handler.py
similarity index 100%
rename from handler.py
rename to src/handler.py
diff --git a/requirements.txt b/src/requirements.txt
similarity index 100%
rename from requirements.txt
rename to src/requirements.txt
diff --git a/start.sh b/src/start.sh
similarity index 100%
rename from start.sh
rename to src/start.sh
diff --git a/test_input.json b/src/test_input.json
similarity index 100%
rename from test_input.json
rename to src/test_input.json
diff --git a/utils.py b/src/utils.py
similarity index 100%
rename from utils.py
rename to src/utils.py