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