Add custom error handling