improvements to predict for non-gaussian models with varying u

improvements to predict for non-gaussian models with varying u #42