Update to the socket demo