diff --git a/Dockerfile b/Dockerfile
index d7eae4d..5fabc64 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,4 +12,4 @@ RUN go mod download
 
 RUN go build -o flow .
 
-CMD ["./flow"]
\ No newline at end of file
+CMD ["sh", "-c", "./flow && while true; do echo 'App is running'; sleep 10; done"]
\ No newline at end of file
diff --git a/docker-compose.yml b/docker-compose.yml
index 7486d93..463aaa0 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -4,7 +4,6 @@ services:
     build:
       context: .
       dockerfile: Dockerfile
-    command: sh -c "while true; do echo 'App is running'; sleep 10; done"
     env_file:
       - .env
     environment:
@@ -16,6 +15,8 @@ services:
       POSTGRES_SSL: "${SSLMODE}"
     depends_on:
       - postgres
+    volumes:
+      - ./app_data:/app/data 
 
   postgres:
     image: postgres:latest