Add a setup.py

Closes #1.