Adding socket.io support