Small changes to plotting.