Add timeout for connection