Yes, please add error handling as well and return an error message if the operation fails.