Updated plot_pspec.py script to work.