adding my e-step for model A. Working on model B e-stp