Replaced getopt with argparse