Update __main__.py