Fixed further comments in main class.