Updated cassandra test with count