Add a small test script