Switch to argparse - allows for nargs