fixed bug in columns + output predictive probs