DataStax interview question

Just coding to fix issue