Apply PEP8 to __main__.py