Add socket shutdown