https://github.com/OpenFreeEnergy/gufe/blob/441b70a6f716038dfa6c139c89c2ce2ebff6db14/gufe/visualization/mapping_visualization.py#L167C1-L173C6 @richardjgowers How about this addition, if we can assume ofe-name is always present?: ``` d2d.DrawMolecules( copies, highlightAtoms=atoms_list, highlightBonds=bonds_list, highlightAtomColors=atom_colors, highlightBondColors=bond_colors, legends=[copies[i].GetProp("ofe-name"), copies[j].GetProp("ofe-name")] ) ``` 
https://github.com/OpenFreeEnergy/gufe/blob/441b70a6f716038dfa6c139c89c2ce2ebff6db14/gufe/visualization/mapping_visualization.py#L167C1-L173C6
@richardjgowers
How about this addition, if we can assume ofe-name is always present?: