Need to update proper way to handle the error handling and exception checking in the notebook environment.