added probability selection on command line