Fix pep8 issues in main.py