Flake8 and add __main__