modfiy argparse & output