Add EM algorithm