Add support for socket.io