Add mode comments to the classes