diff --git a/06-demo-application/load-generator-python/Dockerfile b/06-demo-application/load-generator-python/Dockerfile
index d0a73f7..d6f1b06 100644
--- a/06-demo-application/load-generator-python/Dockerfile
+++ b/06-demo-application/load-generator-python/Dockerfile
@@ -1,13 +1,4 @@
-FROM python:3.12-slim AS generate-requirements
-RUN apt-get update && apt-get install -y curl && rm -rf /var/lib/apt/lists/*
-RUN curl -sSL https://install.python-poetry.org | python3 -
-COPY pyproject.toml poetry.lock ./
-RUN /root/.local/bin/poetry export --output requirements.txt
-
-###
-
-FROM python:3.12-slim AS production
-COPY --from=generate-requirements /requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
-COPY . . 
-CMD ["python", "main.py"]
\ No newline at end of file
+FROM python:3.12-slim
+COPY . .
+RUN pip install --no-cache-dir .
+CMD ["python", "-m", "load_generator_python.main"]
\ No newline at end of file
diff --git a/06-demo-application/load-generator-python/load_generator_python/__init__.py b/06-demo-application/load-generator-python/load_generator_python/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/06-demo-application/load-generator-python/main.py b/06-demo-application/load-generator-python/load_generator_python/main.py
similarity index 100%
rename from 06-demo-application/load-generator-python/main.py
rename to 06-demo-application/load-generator-python/load_generator_python/main.py
diff --git a/06-demo-application/load-generator-python/pyproject.toml b/06-demo-application/load-generator-python/pyproject.toml
index b4eee3e..7f134ce 100644
--- a/06-demo-application/load-generator-python/pyproject.toml
+++ b/06-demo-application/load-generator-python/pyproject.toml
@@ -4,6 +4,10 @@ version = "0.1.0"
 description = ""
 authors = ["sid palas"]
 
+packages = [
+    { include = "load_generator_python" }
+]
+
 [tool.poetry.dependencies]
 python = "^3.9"
 requests = "^2.32.2"