Add argument for running all tests to test script