setting up to work with command line arg.