Minor changes to code for making plots
Minor changes to code for making plots