Minor improvement to visualisation.