from helper import save_screenshot save_screenshot("test")