Code Review comments

Typo in documentation and organising imports