diff --git a/src/routes/viewer.ts b/src/routes/viewer.ts
index 414653d..b53e80a 100644
--- a/src/routes/viewer.ts
+++ b/src/routes/viewer.ts
@@ -81,13 +81,12 @@ router.get(/.*/, async (req: Request, res: Response) => {
                     ${body}
                 </div>
             </body>
+
             <script>
                 window.VIV_PORT = "${config.port}";
                 window.VIV_PATH = "${urlToPath(req.path)}";
             </script>
-
             ${config.scripts ? `<script type="text/javascript">${config.scripts}</script>` : ''}
-
             <script type="module" src="/static/client.mjs"></script>
         </html>
     `);
diff --git a/static/client.mjs b/static/client.mjs
index 8c28889..1235142 100644
--- a/static/client.mjs
+++ b/static/client.mjs
@@ -1,3 +1,6 @@
+/* --------------------------------------------------------------------------
+ * MERMAID ------------------------------------------------------------------ */
+
 import mermaid from '/static/mermaid/mermaid.esm.min.mjs';
 
 const darkModePreference = window.matchMedia('(prefers-color-scheme: dark)');
@@ -17,6 +20,9 @@ try {
     } catch {}
 }
 
+/* --------------------------------------------------------------------------
+ * WEBSOCKET COMMUNICATION WITH SERVER -------------------------------------- */
+
 const ws = new WebSocket(`ws://localhost:${window.VIV_PORT}`);
 
 ws.addEventListener('message', (event) => {
@@ -32,6 +38,7 @@ ws.addEventListener('message', (event) => {
                 await mermaid.run({ querySelector: '.mermaid' });
             })();
             break;
+
         case 'SCROLL':
             let line = parseInt(value);
             while (line) {
@@ -46,9 +53,11 @@ ws.addEventListener('message', (event) => {
                 line -= 1;
             }
             break;
+
         case 'RELOAD':
             window.location.reload();
             break;
+
         case 'PRINT':
             console.log(value);
             break;