All the code works now, except the regression imputation is not happy with lots of binary input. I imagine mean and multiple are sufficient for now.