code: def add_keypress_callback(self, key, fn):
          
          self.viewer.keypress[key].append(fn)
docs: Allows for custom callback functions for the viewer. Called on key down.
          Parameter 'any' will ensure that the callback is called on any key down,
          and block default mujoco viewer callbacks from executing, except for
          the ESC callback to close the viewer.