minor change to the plot