Add the possibility to impute by rows