Add functionality to restore the original volume after disconnecting.