from pyglib.mbody.multiplets_analysis_soc import calc_save_atomic_states,
        plot_atomic_states

calc_save_atomic_states(imp=1, num_ev=100)
plot_atomic_states(imp=1, num_label=5)