Replacing optparse with argparse