Added command-line argument parsing with the argparse module