updated to pass pep8