Updated comment for main function