Improve labelling of output graphs