Added an input argument to the nvdimm script