code now follows pep8