minor tweaks to vds_impute.R