def update(self, arm, reward):
    self.r[arm] = ((self.r[arm] * self.n[arm]) + reward)
    self.n[arm] += 1
    self.r[arm] /= self.n[arm]
    self.t += 1
    self.returns += reward
def update(self, arm, reward):
    self.r[arm] = ((self.r[arm] * self.n[arm]) + reward)
    self.n[arm] += 1
    self.r[arm] /= self.n[arm]
    self.t += 1
    self.returns += reward