Make additional outputs optional

Add argparse with two flags.