Edit to code example