PEP-8 styling on main.py