Add some comments in main function