Update setup.py

Added new script to setup.py