Add connection error handler