Fixed bugs in EM algorithm