Add error validation
Add error message
Bug fix