Update setup.py

Fixing setup.py