Changed few details in data.impute function