Extracting classes to make the interaction with cassandra code base easier to test