Adding Bandit Code