PEP257-ified the code