Switch from socket.io to ws