pep8 editing
Updated