Add error handling v2