pep8 and better docs