ENH: checked with pep8