improving test script