Add code for socket.io server