added probability flag