Finally fix probability
Attribution of some code to JJ