Implement EM for GMM