add new expected improvement function