Improved plotting.

Changed plotting so that the plot has the same size as the previous
plot.