diff --git a/frontend/src/App.tsx b/frontend/src/App.tsx
index 658b6ef..d0e7ee6 100644
--- a/frontend/src/App.tsx
+++ b/frontend/src/App.tsx
@@ -162,7 +162,9 @@ function App() {
     generateCode(
       wsRef,
       updatedParams,
+      // On change
       (token) => setGeneratedCode((prev) => prev + token),
+      // On set code
       (code) => {
         setGeneratedCode(code);
         if (params.generationType === "create") {