Update YCSB script