Add new error checking