added comments to the main class