probs were not probabilities hence causing issues to the dirichlet function