Added support for rocksdb