change some incorrect lines in __main__