diff --git a/example/webui/routes.go b/example/webui/routes.go
index a84f8c9..af47b70 100644
--- a/example/webui/routes.go
+++ b/example/webui/routes.go
@@ -40,7 +40,6 @@ func RegisterRoutes(webapp *fiber.App, pool *AgentPool, db *InMemoryDatabase, ap
 
 	// Define a route for the GET method on the root path '/'
 	webapp.Get("/sse/:name", func(c *fiber.Ctx) error {
-
 		m := pool.GetManager(c.Params("name"))
 		if m == nil {
 			return c.SendStatus(404)
diff --git a/example/webui/views/agents.html b/example/webui/views/agents.html
index aa4014d..067aeef 100644
--- a/example/webui/views/agents.html
+++ b/example/webui/views/agents.html
@@ -5,6 +5,13 @@
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <title>Agent List</title>
     {{template "views/partials/header"}}
+    <style>
+        .button-container {
+            display: flex;
+            justify-content: flex-end;
+            margin-bottom: 8px; /* Adjusts the spacing to your liking */
+        }
+    </style>
 </head>
 <body class="bg-gray-900 p-4 text-white font-sans">
     {{template "views/partials/menu"}}
@@ -12,6 +19,11 @@
         <header class="text-center mb-8">
             <h1 class="text-3xl md:text-5xl font-bold">Smart Agent List</h1>
         </header>
+        <div class="button-container">
+            <a href="/create" class="bg-green-500 hover:bg-green-700 text-white font-bold py-2 px-4 rounded transition duration-300 ease-in-out">
+                Add New Agent
+            </a>
+        </div>
         <section class="flex flex-col">
             <div class="overflow-x-auto">
                 <div class="py-2 align-middle inline-block min-w-full">
@@ -36,9 +48,9 @@
                             <tbody class="bg-gray-800 divide-y divide-gray-700">
                                 <!-- Dynamic agent rows go here -->
                                 {{ range .Agents }}
-                                <tr>
+                                <tr hx-ext="sse" sse-connect="/sse/{{.}}">
                                     <td class="px-6 py-4 whitespace-nowrap text-sm font-medium text-gray-300">{{.}}</td>
-                                    <td class="px-6 py-4 whitespace-nowrap text-sm text-gray-300">{{.}}</td>
+                                    <td class="px-6 py-4 whitespace-nowrap text-sm text-gray-300"><div sse-swap="status" ></div></td>
                                     <td class="px-6 py-4 whitespace-nowrap text-center text-sm font-medium">
                                         <a href="/talk/{{.}}" class="text-indigo-500 hover:text-indigo-400">
                                             <i class="fas fa-comments"></i> Talk
@@ -58,9 +70,7 @@
                 </div>
             </section>
             <footer class="mt-8 text-center">
-                <a href="/create" class="bg-indigo-500 hover:bg-indigo-700 text-white font-bold py-2 px-4 rounded transition duration-300 ease-in-out">
-                    Add New Agent
-                </a>
+                <!-- Moved the button up for alignment with the table -->
             </footer>
         </div>
     </body>
diff --git a/example/webui/views/chat.html b/example/webui/views/chat.html
index aa391c5..92a86f0 100644
--- a/example/webui/views/chat.html
+++ b/example/webui/views/chat.html
@@ -3,13 +3,9 @@
 <head>
     <title>Smart Agent Interface</title>
     {{template "views/partials/header"}}
-    <!-- Optional: Include HTMX and SSE Extension for dynamic updates -->
-    <script src="https://unpkg.com/htmx.org"></script>
-    <script src="https://unpkg.com/htmx.org/dist/ext/sse.js"></script>
-    <script src="https://unpkg.com/hyperscript.org@0.9.12"></script>
     <style>
         body { overflow: hidden; }
-        .chat-container { height: 100vh; display: flex; flex-direction: column; }
+        .chat-container { height: 90vh; display: flex; flex-direction: column; }
         .chat-messages { overflow-y: auto; flex-grow: 1; }
         .htmx-indicator{
                 opacity:0;
diff --git a/example/webui/views/partials/header.html b/example/webui/views/partials/header.html
index 1e7f6d5..cbba1da 100644
--- a/example/webui/views/partials/header.html
+++ b/example/webui/views/partials/header.html
@@ -3,6 +3,9 @@
 <link href="https://cdn.jsdelivr.net/npm/tailwindcss@2.2.19/dist/tailwind.min.css" rel="stylesheet">
 <link href="https://fonts.googleapis.com/css2?family=Roboto:wght@400;500;700&display=swap" rel="stylesheet">
 <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0-beta3/css/all.min.css">
+<script src="https://unpkg.com/htmx.org"></script>
+<script src="https://unpkg.com/htmx.org/dist/ext/sse.js"></script>
+<script src="https://unpkg.com/hyperscript.org@0.9.12"></script>
 <style>
     body {
         font-family: 'Roboto', sans-serif;
