diff --git a/lua/gmod_integration/server/sv_websocket.lua b/lua/gmod_integration/server/sv_websocket.lua
index f463a0e..42e1ac0 100644
--- a/lua/gmod_integration/server/sv_websocket.lua
+++ b/lua/gmod_integration/server/sv_websocket.lua
@@ -6,7 +6,7 @@ if (!gmInte.config.websocket) then return end
 
 require("gwsockets")
 
-if (GWSockets) then return gmInte.logError("GWSockets is not installed! Please install it from https://github.com/FredyH/GWSockets") end
+if (!GWSockets) then return gmInte.logError("GWSockets is not installed! Please install it from https://github.com/FredyH/GWSockets") end
 
 local socket = GWSockets.createWebSocket("wss://ws.gmod-integration.com")
 
@@ -14,40 +14,21 @@ local socket = GWSockets.createWebSocket("wss://ws.gmod-integration.com")
 socket:setHeader("id", gmInte.config.id)
 socket:setHeader("token", gmInte.config.token)
 
-// Dev
-function socket:onMessage(txt)
-    print("Received: ", txt)
-end
-
-function socket:onError(txt)
-    print("Error: ", txt)
-end
-
--- We start writing only after being connected here. Technically this is not required as this library
--- just waits until the socket is connected before sending, but it's probably good practice
 function socket:onConnected()
-    print("Connected to echo server")
-    // say hi
-    socket:write("Hello from GMod!")
-    timer.Simple(1, function()
-        socket:write("ping")
-    end)
-
-    timer.Simple(2, function()
-        timer.Remove("SocketWriteTimer")
-        -- Even if some of the messages have not reached the other side yet, this type of close makes sure
-        -- to only close the socket once all queued messages have been received by the peer.
-        socket:close()
-    end)
+    gmInte.log("WebSocket connected", true)
 end
 
 // log on message
 function socket:onMessage(txt)
-    print("Received: ", txt)
+    gmInte.log("WebSocket Message: " .. txt, true)
 end
 
 function socket:onDisconnected()
-    print("WebSocket disconnected")
+    gmInte.log("WebSocket disconnected", true)
+end
+
+function socket:onError(txt)
+    gmInte.logError("WebSocket Error: " .. txt)
 end
 
 socket:open()
\ No newline at end of file