Fix docs and a little more PEP8.