Add some Cassandra connection code