Add socket stuff