Made some changes to standard run to fix bug with imputation