Updated with MMSE KKT formulation