Correct setup.py
Update setup.py correctly