bandit problem main script done. but its not working