Error handling
Add error handling