diff --git a/Dockerfile b/Dockerfile
index e06728e..13a6f44 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,5 @@
-# Build Stage
-FROM debian:bullseye-slim
+# === Build Stage ===
+FROM debian:bullseye-slim AS builder
 ENV IS_DOCKER TRUE
 
 LABEL org.opencontainers.image.source https://github.com/JustKato/drive-health
@@ -43,6 +43,18 @@ RUN apt-get purge -y gcc musl-dev libsqlite3-dev wget \
     && apt-get autoremove -y \
     && apt-get clean
 
+# === Final Stage ===
+FROM debian:bullseye-slim AS final
+
+# Set the environment variable
+ENV IS_DOCKER TRUE
+
+# Create the directory and set it as the working directory
+WORKDIR /app
+
+# Copy only the necessary files from the builder stage
+COPY --from=builder /app/drive-health .
+
 # Expose the necessary port
 EXPOSE 8080
 
@@ -50,4 +62,4 @@ EXPOSE 8080
 VOLUME [ "/data" ]
 
 # Command to run the executable
-CMD ["./drive-health"]
\ No newline at end of file
+CMD ["./drive-health"]