- Add error handling