Add missing imports to example

[peterjc]

Description

Add some missing imports to help novices.

Checklist

This pull request is:

• [x] A documentation / typographical error fix
  • Good to go, no issue or tests are needed
• [ ] A short code fix
  • please include the issue number, and create an issue if none exists, which must include a complete example of the issue. one line code fixes without an issue and demonstration will not be accepted.
  • Please include: Fixes: #<issue number> in the commit message
  • please include tests. one line code fixes without tests will not be accepted.
• [ ] A new feature implementation
  • please include the issue number, and create an issue if none exists, which must include a complete example of how the feature would look.
  • Please include: Fixes: #<issue number> in the commit message
  • please include tests.

Have a nice day!

[CaselIT]

thanks!