Updated e-steps