Adding probability predictions