Add support for rocksdb