remove extra imports
Cleanup example to post question
Formatting