added bandit code to assn3