diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py
index e96e720..bd4238b 100644
--- a/backend/routes/generate_code.py
+++ b/backend/routes/generate_code.py
@@ -85,6 +85,8 @@ async def stream_code(websocket: WebSocket):
 
     print("Incoming websocket connection...")
 
+
+    ## Communication protocol setup
     async def throw_error(
         message: str,
     ):
@@ -107,6 +109,8 @@ async def stream_code(websocket: WebSocket):
             {"type": type, "value": value, "variantIndex": variantIndex}
         )
 
+    ## Parameter validation
+    
     # TODO: Are the values always strings?
     params: Dict[str, str] = await websocket.receive_json()
     print("Received params")