Modify main_*.py

Add " if __name__ == '__main__': " statement