Added Riak integration