diff --git a/examples/experimental/nodes/frontend/src/App.tsx b/examples/experimental/nodes/frontend/src/App.tsx
index cc4a1eca..cc7d1986 100644
--- a/examples/experimental/nodes/frontend/src/App.tsx
+++ b/examples/experimental/nodes/frontend/src/App.tsx
@@ -168,6 +168,13 @@ const App: React.FC = () => {
         setMessages(prev => [...prev, newMessage]);
         break;
 
+      case "thought":
+        setMessages(prev => [...prev, {
+          text: `💭 ${agentName} is thinking: ${messageData.data.argument}`,
+          type: 'status' as const
+        }]);
+        break;
+
       case "write":
         const filePath = messageData.data.path;
         const fileContent = messageData.data.argument;
diff --git a/examples/experimental/nodes/frontend/src/components/CodeEditor/CodeEditor.tsx b/examples/experimental/nodes/frontend/src/components/CodeEditor/CodeEditor.tsx
index 01ed7b10..54b1b9b5 100644
--- a/examples/experimental/nodes/frontend/src/components/CodeEditor/CodeEditor.tsx
+++ b/examples/experimental/nodes/frontend/src/components/CodeEditor/CodeEditor.tsx
@@ -4,6 +4,7 @@ import { html } from '@codemirror/lang-html';
 import { css } from '@codemirror/lang-css';
 import { python } from '@codemirror/lang-python';
 import { githubDark } from '@uiw/codemirror-theme-github';
+import { EditorView } from '@codemirror/view';
 import './CodeEditor.css'; // Import the CSS file
 
 interface CodeEditorProps {
@@ -51,7 +52,10 @@ const CodeEditor: React.FC<CodeEditorProps> = ({ code, onChange, filename }) =>
           value={fillEmptyLines(code)}
           height="100%"
           theme={githubDark}
-          extensions={getLanguageExtension(filename)}
+          extensions={[
+            ...getLanguageExtension(filename),
+            EditorView.lineWrapping,
+          ]}
           onChange={onChange}
           basicSetup={{
             lineNumbers: true,