Make code samples consistent with the rest of the doc