Adding main and argparse to take args from command line instead of Q&D hard coded category name in code