Added comments in main