Added the main script to setup.py

Added the main script to setup.py