Add information for setup.py

Add information for setup.py