Added e-step