OOP version of bandit
added softmax selection