Change optparse to argparse