diff --git a/Dockerfile b/Dockerfile
index 1d603a1..03817e8 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -39,6 +39,9 @@ RUN apk add --no-cache \
 RUN addgroup -g 1000 appuser && \
     adduser -D -u 1000 -G appuser appuser
 
+# Copy credentials file if it exists
+COPY --chown=appuser:appuser credentials.json .
+
 # Copy virtual environment from builder
 COPY --from=builder /opt/venv /opt/venv
 ENV PATH="/opt/venv/bin:$PATH"
diff --git a/static/js/main.js b/static/js/main.js
index dab34de..51a80ef 100644
--- a/static/js/main.js
+++ b/static/js/main.js
@@ -1,34 +1,35 @@
 (($) => {
-    var ws = new WebSocket("ws://localhost:8000/ws");
-    const input = $("#message");
-    const messages = $('#messages');
-    var lastMessage;
+  var ws = new WebSocket(
+    `ws://${location.protocol + "//" + location.host}${location.port ? ":" + location.port : ""}/ws`,
+  );
+  const input = $("#message");
+  const messages = $("#messages");
+  var lastMessage;
 
-    $('#button').on('click', () => {
-        const message = input.val();
+  $("#button").on("click", () => {
+    const message = input.val();
 
-        if (message) {
-            ws.send(message);
-            lastMessage = $('<div class="message received"><p>Loading...</p></div>')
-            messages.append(`<div class="message sent"><p>${message}</p></div>`)
-            messages.append(lastMessage);
+    if (message) {
+      ws.send(message);
+      lastMessage = $('<div class="message received"><p>Loading...</p></div>');
+      messages.append(`<div class="message sent"><p>${message}</p></div>`);
+      messages.append(lastMessage);
 
-            input.val("");
-        }
-    });
+      input.val("");
+    }
+  });
 
-    ws.onmessage = (event) => {
-        if (lastMessage.text() === "Loading...") {
-            lastMessage.empty();
-        }
+  ws.onmessage = (event) => {
+    if (lastMessage.text() === "Loading...") {
+      lastMessage.empty();
+    }
 
-        var content = document.createTextNode(event.data);
+    var content = document.createTextNode(event.data);
 
-        if (content.textContent === "<<END>>") {
-            lastMessage.html(marked.parse(lastMessage.text()));
-        }
-        else {
-            lastMessage.append(content);
-        }
-    };
+    if (content.textContent === "<<END>>") {
+      lastMessage.html(marked.parse(lastMessage.text()));
+    } else {
+      lastMessage.append(content);
+    }
+  };
 })(jQuery);