Use ws instead of socket.io