minor modification to the plot