Small comment update in MainClass