Add socket timeout