added bandit for testing