fix and improve mohca_plot