probs were not probabilities hence causing issues for the dirichlet function