Update setup.py

Hi:
I've updated the setup.py for #1