Minor changes to plotting code.