add bandits to gridworld