Add argparse and remove file folder