Add riak_cs support (imported from riak_cs/bench)