removing plot_sag_comparison and putting it into plot_sgd_comparison