Fixed the em algorithm