Dirichlet model works better. Also added UCB1 Algorithm for bandit problems.