Add connection timeout mechanism