use argparse
change to use parameter