Update script to take arguments on command line instead of them being hard coded.