pep8, improve docs