Fixes setup.py

The current setup.py doesn't work; here's a fix.

Could we also get this tagged and a package uploaded to pypi?