Got a working version using cassandra