Update example code to be correct

[DominicOram]

To test:

• Import the code into an IDE and confirm it has correct imports etc.

[codecov]

Codecov Report

Merging #277 (eca21f2) into main (14d1bc0) will not change coverage.
The diff coverage is n/a.

@@           Coverage Diff           @@
##             main     #277   +/-   ##
=======================================
  Coverage   89.44%   89.44%           
=======================================
  Files          41       41           
  Lines        1412     1412           
=======================================
  Hits         1263     1263           
  Misses        149      149           

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more

[DominicOram]

Thanks @tpoliaw, what's your workflow here? Are you happy for me to squash and merge?

[tpoliaw]

I think squash and merge is the only strategy enabled. Would be good to remove the To test:.. comment from the auto-generated message though.