updated styling to fit pep8