diff --git a/level_4/Dockerfile b/level_4/Dockerfile
index 8faf4d439..10a796cd6 100644
--- a/level_4/Dockerfile
+++ b/level_4/Dockerfile
@@ -50,4 +50,4 @@ COPY api.py /app
 COPY entrypoint.sh /app/entrypoint.sh
 RUN chmod +x /app/entrypoint.sh
 
-ENTRYPOINT ["/app/cognitive_architecture/entrypoint.sh"]
\ No newline at end of file
+ENTRYPOINT ["/app/entrypoint.sh"]
\ No newline at end of file
diff --git a/level_4/api.py b/level_4/api.py
index 9d5970456..d7f420097 100644
--- a/level_4/api.py
+++ b/level_4/api.py
@@ -88,7 +88,7 @@ async def add_memory(
         )
 
 @app.post("/user-to-graph-query")
-async def generate_cypher_query(payload: Payload,):
+async def generate_cypher_query(payload: Payload):
     try:
 
 
diff --git a/level_4/main.py b/level_4/main.py
index 573c11e35..e0dca6440 100644
--- a/level_4/main.py
+++ b/level_4/main.py
@@ -8,7 +8,7 @@ from cognitive_architecture.database.postgres.models.memory import MemoryModel
 import os
 from dotenv import load_dotenv
 
-from level_4.cognitive_architecture.database.postgres.database_crud import session_scope
+from cognitive_architecture.database.postgres.database_crud import session_scope
 from cognitive_architecture.database.postgres.database import AsyncSessionLocal
 from cognitive_architecture.utils import generate_letter_uuid
 import instructor
diff --git a/level_4/poetry.lock b/level_4/poetry.lock
index 85a868417..aa3c8dd04 100644
--- a/level_4/poetry.lock
+++ b/level_4/poetry.lock
@@ -1564,6 +1564,22 @@ cachetools = "*"
 numpy = "*"
 requests = "*"
 
+[[package]]
+name = "graphviz"
+version = "0.20.1"
+description = "Simple Python interface for Graphviz"
+optional = false
+python-versions = ">=3.7"
+files = [
+    {file = "graphviz-0.20.1-py3-none-any.whl", hash = "sha256:587c58a223b51611c0cf461132da386edd896a029524ca61a1462b880bf97977"},
+    {file = "graphviz-0.20.1.zip", hash = "sha256:8c58f14adaa3b947daf26c19bc1e98c4e0702cdc31cf99153e6f06904d492bf8"},
+]
+
+[package.extras]
+dev = ["flake8", "pep8-naming", "tox (>=3)", "twine", "wheel"]
+docs = ["sphinx (>=5)", "sphinx-autodoc-typehints", "sphinx-rtd-theme"]
+test = ["coverage", "mock (>=4)", "pytest (>=7)", "pytest-cov", "pytest-mock (>=3)"]
+
 [[package]]
 name = "greenlet"
 version = "3.0.1"
@@ -6291,4 +6307,4 @@ testing = ["big-O", "jaraco.functools", "jaraco.itertools", "more-itertools", "p
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.10"
-content-hash = "43d41328cfe0e1eb7cb280efc4fbd4389b56fd71f7c98eb5d1d9e98041d5c8c8"
+content-hash = "9c465587d5f20c3c3837ed0b059f2db3bbc529b328274da37ed92bd6236421c4"
diff --git a/level_4/pyproject.toml b/level_4/pyproject.toml
index 809641db4..b2affdef6 100644
--- a/level_4/pyproject.toml
+++ b/level_4/pyproject.toml
@@ -44,6 +44,7 @@ segment-analytics-python = "^2.2.3"
 pdf2image = "^1.16.3"
 instructor = "^0.3.4"
 networkx = "^3.2.1"
+graphviz = "^0.20.1"