Added argparse to capture script