Fixed docstring location
Fix whitespace, import order and quotes