Tweak the code example