move module imports, fix indentation
docstrings