Add error-checking return