```ruby

module RedditApi
  class Query

    attr_reader :count, :endpoint, :offset_id, :resource_type

    def initialize(args = {})
      @count = args.fetch(:count)
      @endpoint = args.fetch(:endpoint, "")
      @offset_id = args.fetch(:offset, "")
      @records = args.fetch(:records, {})
      @resource_type = args.fetch(:resource, "")
    end

    def add_records(new_records)
      if count > 1
        add_multiple_records(new_records)
      else
        add_single_record(new_records)
      end
    end

    def captured_records
      records.values
    end

    def capture_count
      records.length
    end

    protected
    attr_writer :offset_id
    private
    attr_reader :records

    def add_multiple_records(new_records)
      new_records.each do |record|
        add_single_record(record)
        break if records.length == count
      end
    end

    def add_single_record(record)
      record_id = record_id(record)
      records[record_id] = record
      update_offset_id(record)
    end

    def update_offset_id(record)
      self.offset_id = record_id(record)
    end

    def record_id(record)
      record["data"]["id"]
    end

  end
end
```

Refactor Query to account for records w/ data attr
user
```ruby

module RedditApi
  class Query

    attr_reader :count, :endpoint, :offset_id, :resource_type

    def initialize(args = {})
      @count = args.fetch(:count)
      @endpoint = args.fetch(:endpoint, "")
      @offset_id = args.fetch(:offset, "")
      @records = args.fetch(:records, {})
      @resource_type = args.fetch(:resource, "")
    end

    def add_records(new_records)
      if count > 1
        add_multiple_records(new_records)
      else
        add_single_record(new_records)
      end
    end

    def captured_records
      records.values
    end

    def capture_count
      records.length
    end

    protected
    attr_writer :offset_id
    private
    attr_reader :records

    def add_multiple_records(new_records)
      new_records.each do |record|
        record_id = record["data"]["id"]
        records[record_id] = record
        update_offset_id(record_id)
        break if records.length == count
      end
    end

    def add_single_record(record)
      record_id = record["id"]
      records[record_id] = record
      update_offset_id(record_id)
    end

    def update_offset_id(record_id)
      self.offset_id = record_id
    end

  end
end
```
assistant