CORRECT code to pep8 form

MODIFIED main.py - fix pep8 problems