Add simple error handling