If either of you want this feature pleasure comment on Kedro Viz issue https://github.com/kedro-org/kedro-viz/issues/480