I think generate_prob_list is working :)