Add error handling with test