Changing code example to match documentation