Riak module updates

A little update to the Riak module.