Updated doc to match code