Add generic error handling