Setup script

I added a commit with the changes you need to get this to work with pip.

How do you want to handle this? You can cherry-pick onto your pull request if you want. Or we can merge yours and apply my commit on top.