Added script to setup.py