lots of corrections, probabilities generated look better