Updated EM algorithm given paper derivation