add comments in main