Updated KKT MSE and figures