Either way, I think this class should be moved to `utils.py` in this directory (new file)