add some comments in main function python file