`defaultdict(list)` I think would work here