altered to take a command line argument