Update setVolume method