R code for EM algorithm