Changed up code to possibly support n armed bandits. Tests are not working correctly and I should rewrite starting with unit tests.