Update setup.py

add