Updated the example code

To bring it inline with the tutorial changes
refs DOCS-3410