Update plot_random_effects.R