Update code example to match code