Update impute.h