Converting wordcount script to use argparse