Updated script to accept command line arg
Updated script to accept command line arg