Refactored MAB code to take new models. Created results file to generate plots.