flake8 fixes in main.py (How did this ever pass?)