Mofiied test script