Added modified Bandit_test