def METHOD_NAME(self, main, before=None):
    """
    Handle wait before and after adding screenshot.
    Args:
        main (ModuleBase):
        before (int, float, tuple): Sleep before adding.
    """
    if before is None:
        before = main.config.WAIT_BEFORE_SAVING_SCREEN_SHOT
    if self:
        main.handle_info_bar()
        main.device.sleep(before)
        main.device.screenshot()
        self.add(main.device.image)