Update setup.py

Fix pip install