PEP 8 and docs changes