Add better test script