add comments, pep8