Adds error handling