Add network failure handling