added a setup.py so the package can be installed with `pip`