Cosmetic changes to plot generation script