diff --git a/Dockerfile b/Dockerfile
index c1d34cb..982ccee 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -4,7 +4,7 @@ RUN apk add --no-cache \
     py3-yaml \
     graphviz
 
-WORKDIR /code
-COPY code/ .
+COPY run.sh /run.sh
+RUN chmod +x /run.sh
 
-CMD ["python3", "puudot.py"]
+CMD ["/run.sh"]
diff --git a/docker-compose.yml b/docker-compose.yml
index 62dbc9d..4d2b21c 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -6,5 +6,6 @@ services:
       dockerfile: Dockerfile
     image: puudot:latest
     volumes:
+      - ./code:/code
       - ./data:/data
       - ./config.yaml:/config.yaml
diff --git a/run.sh b/run.sh
new file mode 100644
index 0000000..fb5a466
--- /dev/null
+++ b/run.sh
@@ -0,0 +1,10 @@
+#!/bin/sh
+
+# Check if puudot.py exists
+if [ ! -f "/code/puudot.py" ]; then
+    echo "Error: puudot.py not found in /code directory!"
+    exit 1
+fi
+
+# Run the script
+python3 /code/puudot.py