Update to make socket.io work