Added argparse to run script from command line.