fix example in readme

[crkarthik11]

:memo: What does this PR do ?

What does this implement/fix? Explain your changes.

Updating readme with missing imports

:white_check_mark: Checklist

Before submitting, please make sure you check the following:

• [ X] Lint and tests pass locally with my changes
• [ X] I've added the necessary documentation

Other information