Not entirely sure why this is operating on flattened data, when we had nested dictionaries available natively as part of the class.