Add error handling
Add messages