Add cli arguments to test