Add error handling class