Added comments to classes