Renaming dataset labels.
Replace the assign_nodes vector with a map.