Generate probabilities too :)
Remove old test