Fix PyLint warnings in main.py