argparse improvment