Upper Confidence Bound (UCB) - How to get the dataset