Add error_handling