diff --git a/app.js b/app.js
index 3c8ed9c..30eea35 100644
--- a/app.js
+++ b/app.js
@@ -67,6 +67,7 @@ async function initialize() {
   swarm.on('connection', async (connection, info) => {
     peerCount++;
     updatePeerCount();
+    console.log('Peer connected, current peer count:', peerCount);
 
     // Send the current user's icon to the new peer
     const iconBuffer = await drive.get(`/icons/${userName}.png`);
@@ -94,6 +95,7 @@ async function initialize() {
     connection.on('close', () => {
       peerCount--;
       updatePeerCount();
+      console.log('Peer disconnected, current peer count:', peerCount);
     });
   });
 
@@ -121,7 +123,7 @@ function registerUser(e) {
     reader.onload = (event) => {
       const buffer = new Uint8Array(event.target.result);
       drive.put(`/icons/${regUsername}.png`, buffer);
-      userAvatar = URL.createObjectURL(new Blob([buffer]));
+      userAvatar = `http://localhost:1337/icons/${regUsername}.png`; // Set the correct URL
       registeredUsers[regUsername] = userAvatar;
       localStorage.setItem('registeredUsers', JSON.stringify(registeredUsers));
       continueRegistration(regUsername);
@@ -285,11 +287,10 @@ async function updateIcon(username, avatarBuffer) {
   // This can be adjusted as per your needs
   const userIcon = document.querySelector(`img[src*="${username}.png"]`);
   if (userIcon) {
-    userIcon.src = URL.createObjectURL(new Blob([avatarBuffer]));
+    userIcon.src = `http://localhost:1337/icons/${username}.png`; // Ensure the URL is correct
   }
 }
 
-
 function clearMessages() {
   const messagesContainer = document.querySelector('#messages');
   while (messagesContainer.firstChild) {
@@ -302,4 +303,4 @@ function toggleSetupView() {
   setupDiv.classList.toggle('hidden');
 }
 
-initialize();
\ No newline at end of file
+initialize();