Added comments to class