Add script to setup.py