Add keyspace initialization to connection