Add socket wrapper