Working on sockets, still not quite working