Documentation and PEP8 fixes