Examples: Fixes imports and cleans up code

[manrajgrover]

Description of new feature, or changes

This PR fixes imports, removes unwanted imports and also cleans up code

Checklist

• [x] Your branch is up-to-date with the base branch
• [ ] You've included at least one test if this is a new feature
• [x] All tests are passing

Related Issues and Discussions

None

People to notify

None