modified exp_finite_stationary_diagonal_approximation_all and plot_* to improve speed