update __main__.py