added probability list for wrapper