Add helpers to setup.py