DOC: add missing import
PEP8: fix indentation