Fixing up the module to take a CLI arg