Add mute function