Add an error handling function