Update ycsb.py