Adding mute functionality