Add retry support to Riak