fixed a tiny bug in bandit problem main script