diff --git a/docker/auth-bench/Dockerfile b/docker/auth-bench/Dockerfile
deleted file mode 100644
index b8ec8e9..0000000
--- a/docker/auth-bench/Dockerfile
+++ /dev/null
@@ -1,6 +0,0 @@
-FROM ubuntu:latest
-RUN apt-get update && apt-get install -y curl apache2-utils bc && rm -rf /var/lib/apt/lists/*
-COPY run-ab.sh /usr/local/bin/run-ab.sh
-RUN chmod +x /usr/local/bin/run-ab.sh
-RUN mkdir -p /logs
-CMD ["/usr/local/bin/run-ab.sh"]
\ No newline at end of file
diff --git a/docker/qmk/Dockerfile b/docker/qmk/Dockerfile
new file mode 100644
index 0000000..b9c0ecb
--- /dev/null
+++ b/docker/qmk/Dockerfile
@@ -0,0 +1,35 @@
+# Use an official Python runtime as a parent image
+FROM python:3.9-slim
+
+# Set the working directory in the container
+WORKDIR /usr/src/app
+
+# Install dependencies
+# git, wget, and other necessary tools and libraries
+RUN apt-get update && apt-get install -y \
+    git \
+    build-essential \
+    wget \
+    unzip \
+    && rm -rf /var/lib/apt/lists/*
+
+# Clone QMK Firmware repository
+RUN git clone --depth 1 https://github.com/qmk/qmk_firmware.git
+
+# Install QMK dependencies
+RUN cd qmk_firmware && ./util/qmk_install.sh
+
+# Create directories for data and output
+RUN mkdir /usr/src/app/data /usr/src/app/bin
+
+# Add a script that will be executed at runtime
+COPY compile_firmware.sh /usr/src/app
+
+# Make the script executable
+RUN chmod +x /usr/src/app/compile_firmware.sh
+
+# Set the script to run on container startup
+ENTRYPOINT ["/usr/src/app/compile_firmware.sh"]
+
+# Volume for input and output
+VOLUME ["/usr/src/app/data", "/usr/src/app/bin"]