add comments and pep8 compliance