Added argparse for command-line argument support