modified plot_elm_comparison to use RandomLayer instead of RandomHiddenLayer