Some extra features on top of Bandit