ENH: fix to doc test
MAINT: pep8 fixes
MAINT: pep8 fixes