Make test script more useful