added probability calculations