Update riak.py