Consider using a ```collections.defaultdict(list)```