Adding script to setup.py