Tiny changes in the example