Updates for KEfED model