Changing script to use command line arguments again.