Some changes, I think to specifying colours of nodes