- added some comments to main class