Some more bandit logic