More work on bandits part of paper