Added comments to main class