Fixing writing code example