Update file to allow command line arguments