added more probability tracking