Found bug with the size probabilities, working on it.