Changes to plot_rag.py