Update setup.py so that it works now