Updated the driver - input file as arg
Updated the driver - input file as arg