Update socket brrr

README.md

@@ -10,7 +10,7 @@
 local ws = require(game.ServerScriptService:WaitForChild("WebSocket"))
 local server = ws({
 	url = "wss://echo.websocket.org/",
-	onMessage = function(message)
+	onMessage = function(socket, message)
 		print("Message received: " .. message)
 	end,
 	onError = function(error)
@@ -25,8 +25,9 @@ server.sendMessage("Hello World")
 ```lua
 local server = ws({
 	url = "wss://echo.websocket.org/",
-	onMessage = function(message)
+	onMessage = function(socket, message)
 		print("Message received: " .. message)
+		socket.close(); -- Close the socket after receiving a message, as an example
 	end,
 	onError = function(error)
 		print("Error: " .. error)

WebSocket.lua

@@ -20,7 +20,7 @@ end
 local ws = function (dict)
 	local socket = {}
 	local closed = false
-	local onMessage = dict.onMessage or function(msg)
+	local onMessage = dict.onMessage or function(socket, msg)
 		print(msg)
 	end
 	local onError = dict.onError or function(err)