Update memory test script