Remove unused code from example

To avoid confusion