Extension of YCSB