Add error handling
Bug fix