refine __main__.py