Change to argparse module.