Updated tests to use the new arg() function for command line args