Replace optparse with argparse