diff --git a/Dockerfile b/Dockerfile
index 95c77b6..9ee0fe1 100755
--- a/Dockerfile
+++ b/Dockerfile
@@ -16,4 +16,4 @@ RUN python3 -m pip install -e .
 RUN python3 -m pip install -r requirements.txt
 
 # Run the service loop.
-CMD ["python", "mini_queue/run.py"]
+CMD ["python", "src/serve.py"]
diff --git a/scripts/mock_knock.py b/scripts/mock_client.py
similarity index 100%
rename from scripts/mock_knock.py
rename to scripts/mock_client.py
diff --git a/mini_queue/run.py b/src/serve.py
similarity index 96%
rename from mini_queue/run.py
rename to src/serve.py
index 319ae64..fbc73e9 100644
--- a/mini_queue/run.py
+++ b/src/serve.py
@@ -1,13 +1,15 @@
 import logging
 from multiprocessing import Process
+
 import pika
+from flask import Flask, jsonify
 from waitress import serve
-from flask import Flask, request, jsonify
 
 from mini_queue.utils.config import CONFIG
 from mini_queue.utils.rabbitmq import make_channel, declare_queue, make_callback, read_connection_params
 
 
+# TODO: implement meaningful checks
 def start_integrity_checks_webserver(mode="debug"):
 
     app = Flask(__name__)