model robustness greatly improved and avisualize adjustments for plot_latent