Adding optional error handling