added comments to classes