add pep8 and formatting