made the code pep8 compliant