Add updated test script