Add count to db_driver