Use argparse for arguments