minor changes on the plot