added multi-armed bandit test