Added MAB simulation with plotting functions.