Switch to argparse