Re-implemented the function for making the model matrices column stochastic.