I revised generate_prob_list