Minor modification on example