add in setup.py