Added code for calculating the UCB and regret in a bandit game