Added error highlight to imputation