added the MoE model