Fixed some comments in the main