Adding a setup.py, so this can be installed via requirements.txt