N-arm bandit program fixed