Adding disco function