Small changes to plotting.
Now creating more plots.