Added code for selecting probabilities based on paper keywords