Export utilities to separate modules
Support multiple keys in count